[프로그래머스(Java)] 이차원 배열 대각선 순회하기

 

Level.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]의 길이

 

 

풀이 1.

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;
 
    }
}

 

 

풀이 2. - Stream

import java.util.stream.IntStream;

class Solution {
    public int solution(int[][] board, int k) {
        return IntStream.range(0, board.length)
                    .filter(i -> i<=k)
                    .map(i -> IntStream.range(0, board[i].length)
                                .filter( j -> j + i <= k)
                                .map(j -> board[i][j])
                                .sum()
                        )
                    .sum();
        
    }
}

---

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

프로그래머스