[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)

입출력 예시

입출력 예 #1

• 입력: [[0, 1], [2, 5], [3, 9]]
• 출력: 2

입출력 예 #2

• 입력: [[-1, 1], [1, 3], [3, 9]]
• 출력: 0

입출력 예 #3

• 입력: [[0, 5], [3, 9], [1, 10]]
• 출력: 8

풀이

선분이 지나가는 점들을 활용하여 겹치는 부분의 길이를 구한다.

선분 길이 = 점 개수 - 1

겹치는 선 길이 = 겹치는 점 개수

코드

class Solution {
    fun solution(lines: Array<IntArray>): Int {
        // 점의 범위가 -100 <= a < b <= 100 이므로 총 200+1(0)개 점을 가지는 배열을 만든다.
        val count = Array(201) { 0 }
        // 선분이 지나가는 모든 점들을 표시한다.
        lines.forEach {
            // 선분 길이는 시작점부터 끝점까지 존재하는 전체 점 개수에서 1개를 뺀 값이다
            // 그러므로 반복문에서 끝점은 제외한다.
            for (i in it[0] until it[1]) count[i + 100]++
        }
        // 겹치는 점 개수 = 겹치는 선분 길이
        return count.count { it >= 2 }
    }
}

테스트 1 〉	통과 (0.06ms, 60.2MB)
테스트 2 〉	통과 (0.10ms, 60.7MB)
테스트 3 〉	통과 (0.05ms, 61.3MB)
테스트 4 〉	통과 (0.06ms, 61MB)
테스트 5 〉	통과 (0.05ms, 61MB)
테스트 6 〉	통과 (0.05ms, 60.4MB)
테스트 7 〉	통과 (0.05ms, 61.5MB)
테스트 8 〉	통과 (0.08ms, 61.9MB)
테스트 9 〉	통과 (0.05ms, 59.2MB)
테스트 10 〉	통과 (0.07ms, 58.2MB)