[프로그래머스/Lv.0] 평행 | 파이썬

문제

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

점 네 개의 좌표를 담은 이차원 배열  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], [3, 8]을 잇고 [9, 2], [11, 6]를 이으면 두 선분은 평행합니다.

입출력 예 #2

• 점을 어떻게 연결해도 평행하지 않습니다.

 

나의 풀이

from itertools import permutations

def solution(dots):
    permutation = permutations([0, 1, 2, 3], 2)
    a, b = 0, 0
    
    for p1 in permutation: 
        n1, m1 = p1
        n2, m2 = [x for x in [0, 1, 2, 3] if x not in p1]
        
        a = (dots[n1][0] - dots[m1][0]) / (dots[n1][1] - dots[m1][1])
        b = (dots[n2][0] - dots[m2][0]) / (dots[n2][1] - dots[m2][1])
        
        if a == b:
            return 1
                
    return 0

`permutations`를 이용해 4개의 점 중 2개를 고르는 모든 순서쌍을 만들고,
남은 두 점을 또 하나의 쌍으로 구성해 기울기를 구하는 공식을 이용해 비교했다.

 

다른 사람의 풀이

def solution(dots):
    (x1, y1), (x2, y2), (x3, y3), (x4, y4) = dots
    res1 = (x1 - x2) / (y1 - y2) == (x3 - x4) / (y3 - y4)
    res2 = (x1 - x3) / (y1 - y3) == (x2 - x4) / (y2 - y4)
    res3 = (x1 - x4) / (y1 - y4) == (x2 - x3) / (y2 - y3)
    return 1 if res1 or res2 or res3 else 0

평행 비교를 위해서는 하나의 점을 기준으로 세 가지 경우만 비교하면 되는 것이었다..!