프로그래머스[코딩 기초 트레이닝]_이차원 배열 대각선 순회하기

👾문제 설명

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]의 길이

 

🌌입출력 예

board	k	result
[[0, 1, 2],[1, 2, 3],[2, 3, 4],[3, 4, 5]]	2	8

입출력 예 #1

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

i\j	0	1	2
0	0	1	2
1	1	2	3
2	2	3	4
3	3	4	5

• 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[0])):
            if i+j<=k:
                answer+=board[i][j]
    return answer

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

🤓공부

첫 번째 코드에서는 range(len(board[0]))를 사용하여 board[0]의 길이에 따라 두 번째 for 루프의 범위가 결정된다.

이 경우 board의 각 행의 길이가 동일한 경우에만 정상적으로 동작한다.

만약 board의 행의 길이가 다르다면 오류가 발생할 수 있다.
두 번째 코드에서는 range(len(board))를 사용하여 board의 길이에 따라 두 번째 for 루프의 범위가 결정된다.

이 경우 board의 각 행의 길이가 다르더라도 정상적으로 동작한다.
따라서 첫 번째 코드는 board의 각 행의 길이가 동일할 때 사용하고, 두 번째 코드는 board의 각 행의 길이가 다를 수 있을 때 사용하는 것이 좋다.