[프로그래머스/파이썬] Lv.0 이차원 배열 대각선 순회하기 (11')

Programmers 레벨 0_코딩 기초 트레이닝 문제 풀이

https://school.programmers.co.kr/learn/challenges/training?order=recent

코딩 기초 트레이닝 | 프로그래머스 스쿨

 

 

풀이 날짜 : 2023-09-21

소요 시간 : 11min

 

---

 

문제 설명

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

 

 

 

나의 풀이

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

 

 

 

다른 풀이

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