🐍python3/Programmers

프로그래머스[코딩테스트 입문]_평행

박필중 2023. 6. 4. 09:43
728x90

📃문제설명

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

 

🪧입출력 예

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

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

 

💻코드

def solution(dots):
    s1 = (dots[1][1] - dots[0][1]) / (dots[1][0] - dots[0][0])
    s2 = (dots[2][1] - dots[3][1]) / (dots[2][0] - dots[3][0])
    s3 = (dots[2][1] - dots[0][1]) / (dots[2][0] - dots[0][0])
    s4 = (dots[3][1] - dots[0][1]) / (dots[3][0] - dots[0][0])
    s5 = (dots[2][1] - dots[1][1]) / (dots[2][0] - dots[1][0])
    s6 = (dots[3][1] - dots[1][1]) / (dots[3][0] - dots[1][0])
    
    if s1 == s2 or s3 == s6 or s4 == s5:
        return 1
    else:
        return 0

 

더보기

🧑‍💻공부

기울기를 구하는 방법에 대해 생각해보면 풀 수 있다.

다만 코드가 난잡해 보여 이후 좋은 아이디어가 있으면 수정을 해봐야겠다.

728x90