이차원 배열 대각선 순회하기

송용진·2023년 8월 27일
0

알고리즘

목록 보기
87/173

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

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

내 코드

def solution(board, k):    
    L = []
    for i in range(len(board)):
        for j in range(len(board[i])):
            if i+j<=k:
                L.append(board[i][j])
            else:
                continue
    answer = sum(L)
    return answer

예시 코드

def solution(board, k):
    answer = 0
    n, m = len(board), len(board[0])

    for i in range(n):
        for j in range(m):
            if i + j <= k:
                answer += board[i][j]
            else:
                break
    return answer
profile
백엔드 개발자

0개의 댓글