[파이썬] 코딩테스트연습 <이차원 배열 대각선 순회하기>

문제 설명

2차원 정수 배열 board와 정수 k가 주어집니다.
i + j <= k를 만족하는 모든 (i, j)에 대한 board[i][j]의 합을 return 하는 solution 함수를 완성해 주세요.

 

문제 예시에 글로 되어 있는 문제 설명이 표로 만들어져 있습니다.

 

 

i행의 j열을 하나씩 순서대로 확인해주는 코드를 만들어줍니다.

 

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

 

i행을 먼저 둘러보고, j열을 둘러보도록 for문을 사용한 뒤 if에서 조건을 만족하는 값을 answer에 더해줍니다.