[프로그래머스] 겹치는 선분의 길이

문제 설명

선분 3개가 평행하게 놓여 있습니다. 

세 선분의 시작과 끝 좌표가 [[start, end], [start, end], [start, end]] 형태로 들어있는 2차원 배열 lines가 매개변수로 주어질 때, 두 개 이상의 선분이 겹치는 부분의 길이를 return 하도록 solution 함수를 완성해 보세요.

lines가 [[0, 2], [-3, -1], [-2, 1]]일 때 그림으로 나타내면 다음과 같습니다.

선분이 두 개 이상 겹친 곳은 [-2, -1], [0, 1]로 길이 2만큼 겹쳐있습니다.

제한사항

lines의 길이 = 3

lines의 원소의 길이 = 2

모든 선분은 길이가 1 이상입니다.

lines의 원소는 [a, b] 형태이며, a, b는 각각 선분의 양 끝점입니다.

-100 ≤ a < b ≤ 100

입출력 예

Input : [[0, 1], [2, 5], [3, 9]] / Output : 2
Input : [[-1, 1], [1, 3], [3, 9]] / Output : 0
Input : [[0, 5], [3, 9], [1, 10]] / Output : 8

입출력 예 설명

입출력 예 #1

두 번째, 세 번째 선분 [2, 5], [3, 9]가 [3, 5] 구간에 겹쳐있으므로 2를 return 합니다.

입출력 예 #2

겹친 선분이 없으므로 0을 return 합니다.

입출력 예 #3

첫 번째와 두 번째 선분이 [3, 5] 구간에서 겹칩니다.

첫 번째와 세 번째 선분 [1, 5] 구간에서 겹칩니다.

두 번째와 세 번째 선분 [3, 9] 구간에서 겹칩니다.

따라서 [1, 9] 구간에 두 개 이상의 선분이 겹쳐있으므로, 8을 return 합니다.

제출코드

1. 주어진 배열을 돌아가며 start와 end 사이를 Map에 카운트합니다. 쉽게 생각해서 바둑판에 주사위를 쌓는다고 생각하면 됩니다.
2. Map에 카운트한 개수가 1보다 클 때 겹치는 부분이라고 할 수 있습니다.

여기서 주의할 점은 처음에 rangeClosed() 메서드를 사용해서 forEach를 실행해 보니 기댓값에 2배가 반환되었습니다.

겹치는 점의 개수가 아닌 선의 개수를 반환해야 하기 때문에 rangeClosed()가 아닌 range() 메서드를 사용해 end - 1까지만 forEach를 실행해서 겹치는 선의 개수를 반환해야 합니다.

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.stream.IntStream;

class Solution {
    public int solution(int[][] lines) {
        Map<Integer, Integer> lineMap = new HashMap<>();
        Arrays.stream(lines)
            .forEach(line -> IntStream.range(line[0], line[1])
                .forEach(i -> lineMap.put(i, lineMap.getOrDefault(i, 0) + 1)));
        return (int) lineMap.values().stream().filter(i -> i > 1).count();
    }
}

---

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/120876

프로그래머스