[프로그래머스] Lv.0 평행 [Python]

 

 

4개의 점 중에서 2개씩 이어서 평행이 되는지 확인하는 문제

 

문제 설명
점 네 개의 좌표를 담은 이차원 배열  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축과 평행한 경우는 주어지지 않습니다.

 

처음에는 $\text{기울기} = \frac{y \text{증가율}}{x \text{증가율}} = \frac{y2 - y1}{x2 - x1}$
위 식을 대입해서 풀었는데, 아마도 연산에서 오차가 발생해서 문제가 나는 것 같았습니다.

그래서 $\frac{A}{B}=\frac{C}{D}$ >>> $A \times D = B \times C$를 이용해 정리했습니다.

 

def solution(dots):
    s = []
    a,b,c,d = dots
    m1 = (a[1] - b[1]) * (c[0] - d[0]) == (a[0] - b[0]) * (c[1] - d[1])
    m2 = (a[1] - c[1]) * (b[0] - d[0]) == (a[0] - c[0]) * (b[1] - d[1])
    m3 = (a[1] - d[1]) * (b[0] - c[0]) == (a[0] - d[0]) * (b[1] - c[1])
    return 1 if m1 or m2 or m3 else 0

 

 

 

입출력 예

dots	result
[[1, 4], [9, 2], [3, 8], [11, 6]]	1
[[3, 5], [4, 1], [2, 4], [5, 10]]	0

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

 

입출력 예 #2
점을 어떻게 연결해도 평행하지 않습니다.