https://school.programmers.co.kr/learn/courses/30/lessons/120875
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
문제
점 네 개의 좌표를 담은 이차원 배열 dots가 다음과 같이 매개변수로 주어집니다.
[[x1, y1], [x2, y2], [x3, y3], [x4, y4]]
주어진 네 개의 점을 두 개씩 이었을 때, 두 직선이 평행이 되는 경우가 있으면 1을 없으면 0을 return 하도록 solution 함수를 완성해보세요.
제출 답안
기울기가 같으면 평행
def solution(dots):
dot1 = dots[0]
dot2 = dots[1]
dot3 = dots[2]
dot4 = dots[3]
if (dot1[1]-dot2[1])/(dot1[0]-dot2[0]) == (dot3[1]-dot4[1])/(dot3[0]-dot4[0]):
return 1
if (dot1[1]-dot3[1])/(dot1[0]-dot3[0]) == (dot2[1]-dot4[1])/(dot2[0]-dot4[0]):
return 1
if (dot1[1]-dot4[1])/(dot1[0]-dot4[0]) == (dot2[1]-dot3[1])/(dot2[0]-dot3[0]):
return 1
return 0
개선 답안
- 리스트 언패킹
- 논리값(True/False) 저장 후 한번에 if문 처리
def solution(dots):
[[x1, y1], [x2, y2], [x3, y3], [x4, y4]]=dots
answer1 = ((y1-y2)*(x3-x4) == (y3-y4)*(x1-x2))
answer2 = ((y1-y3)*(x2-x4) == (y2-y4)*(x1-x3))
answer3 = ((y1-y4)*(x2-x3) == (y2-y3)*(x1-x4))
return 1 if answer1 or answer2 or answer3 else 0
참고 자료