[프로그래머스] Lv.0 /이차원 배열 대각선 순회하기 (파이썬/Python)

1. 문제 설명

2. 풀이 과정 

board의 길이len(board)만큼 인덱스 i를 반복하고 board의 열의 길이len(board[0])만큼 인덱스 j를 반복한 다음 i+j의 값이 k보다 작거나 같을 때 board[i][j]의 값을 sum()으로 더해서 반환한다.

3. 코드

def solution(board, k):
    return sum(board[i][j] for i in range(len(board)) for j in range(len(board[0])) if i + j <= k)