[Kotlin] 프로그래머스 - 평행

문제

링크

점 네 개의 좌표를 담은 이차원 배열 dots가 다음과 같이 매개변수로 주어집니다.

[[x1, y1], [x2, y2], [x3, y3], [x4, y4]]

주어진 네 개의 점을 두 개씩 이었을 때, 두 직선이 평행이 되는 경우가 있으면 1을 없으면 0을 return 하도록 solution 함수를 완성해보세요.

제한사항

dots의 길이 = 4

dots의 원소는 [x, y] 형태이며 x, y는 정수입니다. (0 ≤ x, y ≤ 100)

서로 다른 두개 이상의 점이 겹치는 경우는 없습니다.

두 직선이 겹치는 경우(일치하는 경우)에도 1을 return 해주세요.

임의의 두 점을 이은 직선이 x축 또는 y축과 평행한 경우는 주어지지 않습니다.

입출력 예시

입출력 예 #1

• 입력: [[1, 4], [9, 2], [3, 8], [11, 6]]
• 출력: 1
• 설명:  점 [1, 4], [3, 8]을 잇고 [9, 2], [11, 6]를 이으면 두 선분은 평행합니다.

입출력 예 #2

• 입력: [[3, 5], [4, 1], [2, 4], [5, 10]]
• 출력: 0
• 설명: 점을 어떻게 연결해도 평행하지 않습니다.

풀이

평행한지 알려면 점을 이어 생성되는 선이 가지는 기울기를 계산해야 한다.

• 기울기 = y값 증가량 / x값 증가량

두 직선이 겹쳐도 평행한 것으로 판단하기 때문에 y절편을 고려하지 않는다.

평행해야 하므로 각 점들이 연결되지 않도록 한다.

평행한 선이 존재하는 지만 확인하면 되기 때문에 기울기가 같은 선분이 있으면 계산을 종료한다.

코드

class Solution {
    fun solution(dots: Array<IntArray>): Int {
        // 평행이 존재하면 1, 존재하지 않으면 0
        var answer = 0

        // 기울기 기록 - key: 기울기, value: 점1 좌표, 점2 좌표
        val slopes = mutableMapOf<Double, Pair<IntArray, IntArray>>()

        // 동일한 점끼리 비교하지 않도록 반복문 설정
        for (i in 0 until dots.size - 1) {
            if (answer == 1) break
            // 이미 계산된 i번째 좌표는 제외한다.
            for (j in i + 1 until dots.size) {
                // 기울기 = y값 증가량 / x값 증가량
                val x = dots[i][0] - dots[j][0]
                val y = dots[i][1] - dots[j][1]
                val slope = y.toDouble() / x.toDouble()

                // 동일한 기울기 없으면 기록
                if (!slopes.containsKey(slope)) slopes[slope] = Pair(dots[i], dots[j])
                else {
                    slopes.filter { it.key == slope }
                        .forEach {
                            // 각 점들이 연결되는 경우 제외
                            if (dots[i][0] == it.value.first[0] || dots[i][1] == it.value.first[1]) return@forEach
                            if (dots[i][0] == it.value.second[0] || dots[i][1] == it.value.second[1]) return@forEach
                            if (dots[j][0] == it.value.second[0] || dots[j][1] == it.value.second[1]) return@forEach
                            if (dots[j][0] == it.value.first[0] || dots[j][1] == it.value.first[1]) return@forEach
                            answer = 1
                        }
                }
            }
        }

        return answer
    }
}