문제 링크

정답 :

def solution(board, k):
    check = []
    
    for i in range(len(board)):
        for j in range(len(board[0])):
            if i + j <= k:
                check.append(board[i][j])
            else:
                break;
    
    return sum(check)

개선 :

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]
            else:
                break;
    
    return answer