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

문제

선분 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

입출력 예

lines	result
[[0, 1], [2, 5], [3, 9]]	2
[[-1, 1], [1, 3], [3, 9]]	0
[[0, 5], [3, 9], [1, 10]]	8

문제 출처

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

생각

• 선분을 길이 1단위로 나눠서 생각해, 그 시작점을 count하는 Map을 만들자
  • 예를들어, [-3,-1] 의 선분이 있다면, -3, -2가 출현한 횟수를 세고, [1,8] 의 선분은 1, 2, 3, ...7 이 출현한 횟수를 센 배열 map을 만들자
• map의 value가 1보다 클 경우의 수를 cnt

코드

import java.util.HashMap;
import java.util.Map;

class Solution {
    public int solution(int[][] lines) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int[] line : lines) {
            for (int i = line[0]; i < line[1]; i++) {
                map.put(i, map.getOrDefault(i, 0) + 1);
            }
        }
        int cnt = 0;
        for (int key : map.keySet()) {
            if (map.get(key) > 1) {
                cnt++;
            }
        }
        return cnt;
    }
}

다른 사람의 풀이

import java.util.*;
class Solution {
    public int solution(int[][] lines) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int[] line : lines) {
            int from = Math.min(line[0], line[1]);
            int to = Math.max(line[0], line[1]);
            for (int i = from; i < to; i++) {
                map.merge(i, 1, Integer::sum);
            }
        }

        return (int) map.values().stream().filter(i -> i > 1).count();
    }
}

• map.merge()를 이용해서 map에 i라는 키가 있으면, 그 값에 1을 더하고, 없으면 새로운 키에 1을 저장하는 방법으로 구현한 풀이였다
• 또한 1초과인 값을 가진 key의 개수를 스트림으로 간결하게 푼 점도 좋았다! 🙂