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

문제 설명

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

---

풀이

class Solution {
    fun solution(lines: Array<IntArray>): Int {
        var answer: Int = 0
        val duplicate: MutableMap<Pair<Int, Int>, Int> = mutableMapOf()

        lines.forEach {
            for (i in it[0] until it[1])
                duplicate[Pair(i, i + 1)] = duplicate.getOrDefault(Pair(i, i + 1), 0) + 1
        }
        answer = duplicate.filter { it.value >= 2 }.values.size
        return answer
    }
}

처음엔 좌표를 이용해서 풀다가 

[1,2],[2,3],[3,4] 의 경우에 길이는 0인데 1이 나오는 오류가 났다..

고민고민하다가 길이가 1인 구간을 key로 두고 풀었다.

 

다 풀고 좌표로 할때는 어떻게 해결할까 고민하다

마지막 인덱스는 열린구간 즉, count하지 않는 방법을 이용하면 풀릴 것 같다

 

문제 링크