[프로그래머스] Lv.0 - 이차원 배열 대각선 순회하기.py

문제

• 2차원 정수 배열 board와 정수 k가 주어집니다.

• i + j <= k를 만족하는 모든 (i, j)에 대한 board[i][j]의 합을 return 하는 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ board의 길이 ≤ 100
• 1 ≤ board[i]의 길이 ≤ 100
  • 1 ≤ board[i][j] ≤ 10,000
  • 모든 board[i]의 길이는 같습니다.
• 0 ≤ k < board의 길이 + board[i]의 길이

입출력 예시

• 입출력 예 1
  • 입출력 예 #1의 board를 표로 나타내면 다음과 같습니다.

• i + j가 2보다 작거나 같은 항들의 합은 0 + 1 + 2 + 1 + 2 + 2 = 8이므로 8을 return 합니다.

코드

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

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges