[프로그래머스] Lv.0 이차원 배열 대각선 순회하기.java

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

class Solution {
    public int solution(int[][] board, int k) {
        int sum = 0;

        for (int i = 0; i <= k; i++) {
            for (int j = 0; j <= k - i; j++) {
                if (i < board.length && j < board[i].length) {
                    sum += board[i][j];
                }
            }
        }

        return sum;
    }
}

변수 sum을 0으로 초기화합니다. 이 변수는 조건을 만족하는 모든 board[i][j]의 합을 저장할 변수입니다.
중첩된 반복문을 사용하여 조건 i + j <= k를 만족하는 (i, j)에 대해서 board[i][j]의 값을 sum에 더합니다.
외부 반복문은 i를 0부터 k까지 증가시킵니다.
내부 반복문은 j를 0부터 k - i까지 증가시킵니다.
조건 i < board.length와 j < board[i].length를 사용하여 board 배열의 범위를 초과하지 않도록 합니다.
sum 값을 반환합니다.

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges