[프로그래머스] Lv.0 겹치는 선분의 길이 - 자바(Java)

문제: 코딩테스트 연습 - 겹치는 선분의 길이 | 프로그래머스 스쿨 (programmers.co.kr)

문제설명

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

---

나의 문제 풀이

class Solution {
    public int solution(int[][] lines) {
        int answer = 0;
        int[] count = new int[201]; 
        for (int[] i : lines) {
            int start = i[0] + 100; //음수 + 양수를 count에 표현하기 위해서 100을 더함
            int end = i[1] + 100;
            for (int j = start; j < end; j++) count[j]++;
        }
		
        for (int i : count) if(i > 1) answer++;
        
        return answer;
    }
}

---

다른 사람의 문제풀이

//다른사람풀이1 (List, ArrayList 이용)
import java.util.List;
import java.util.ArrayList;
class Solution {
    public int solution(int[][] lines) {
        List<Integer> overLap = new ArrayList<>();
        int cnt = 0;
        for (int i = 0; i <= 2; i++) {
            int s = lines[i][0] > lines[i][1] ? lines[i][1] : lines[i][0];
            int b = lines[i][0] > lines[i][1] ? lines[i][0] : lines[i][1];
            for (int j = s; j < b; j++) {
                if (overLap.contains(j)) {
                    overLap.remove(Integer.valueOf(j));
                    cnt++;
                } else overLap.add(j);
            }
        }
        return cnt;
    }
}

//다른사람풀이2 (Map, HashMap, Math 이용)
import java.util.HashMap;
import java.util.Map;
class Solution {
    public int solution(int[][] lines) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i=0; i<lines.length; i++) {
            int min = Math.min(lines[i][0], lines[i][1]);
            int max = Math.max(lines[i][0], lines[i][1]);
            for (int j=min; j<max; j++) {
                map.put(j, map.getOrDefault(j, 0) + 1);
            }
        }
        int answer = 0;
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            if (entry.getValue() >= 2) {
                answer++;
            }
        }
        return answer;
    }
}

---

느낀점

List와 Map을 이용한 방법으로도 다시 한번 풀어봐야겠다.

---

문제 출처: 코딩테스트 연습 | 프로그래머스 스쿨 (programmers.co.kr)