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