[코딩테스트 Lv.0] 이차원 배열 대각선 순회하기

문제 설명

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 합니다.

---

내 풀이

Java

class Solution {
    public int solution(int[][] board, int k) {
        int answer = 0;
        for(int i = 0; i < board.length; i++){
            for(int j = 0; j < board[i].length; j++){
                if(i + j <= k) answer += board[i][j];
            }
        }
        return answer;
    }
}

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

// board_rows는 2차원 배열 board의 행 길이, board_cols는 2차원 배열 board의 열 길이입니다.
int solution(int** board, size_t board_rows, size_t board_cols, int k) {
    int answer = 0;
    int i = 0, j = 0;
    
    for(i = 0; i < board_rows; i++){
        for(j = 0; j < board_cols; j++){
            if(i + j <= k) answer += board[i][j];
        }
    }    
    return answer;
}

Python

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

---

https://school.programmers.co.kr/learn/courses/30/lessons/181829

프로그래머스