[프로그래머스] 평행 / python

문제

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축과 평행한 경우는 주어지지 않습니다.

 

아이디어

두 직선이 평행하기 위한 조건을 이용한다.

네 개의 점이 만드는 가능한 직선을 모두 조사해서 평행을 조사함.

 

풀이

평행 조건을 검사하는 함수 cal을 작성한다.

점 4개가 만들 수 있는 모든 직선 3개의 경우 모두를 조사한 후 하나라도 평행이 나오면 1을 return.

def solution(dots):
    answer = []
    answer.append(cal(dots[0],dots[1],dots[2],dots[3]))
    answer.append(cal(dots[0],dots[2],dots[1],dots[3]))
    answer.append(cal(dots[0],dots[3],dots[2],dots[1]))
    return 0 if sum(answer) == 0 else 1
def cal(a,b,c,d):
    answer = (a[0]-b[0])*(c[1]-d[1]) - (a[1]-b[1])*(c[0]-d[0])
    return 1 if answer == 0 else 0