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

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

프로그래머스

---

문제

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

---

풀이

import java.util.stream.IntStream;

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

 

 

 

---

GitHub: https://github.com/gajicoding/coding_test/edit/main/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4/0/181829.%E2%80%85%EC%9D%B4%EC%B0%A8%EC%9B%90%E2%80%85%EB%B0%B0%EC%97%B4%E2%80%85%EB%8C%80%EA%B0%81%EC%84%A0%E2%80%85%EC%88%9C%ED%9A%8C%ED%95%98%EA%B8%B0

---

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

코딩테스트 연습 | 프로그래머스 스쿨

---