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)
def is_parallel(a, b, c, d):
x, y = 0, 1
g1 = (b[y] - a[y]) / (b[x] - a[x])
g2 = (d[y] - c[y]) / (d[x] - c[x])
return 1 if g1 == g2 else 0
def solution(dots):
dots.sort(key=lambda x: x[0])
answer = 0
a, b, c, d = dots
aa = is_parallel(a, b, c, d)
bb = is_parallel(a, c, b, d)
cc = is_parallel(a, d, b, c)
if 1 in [aa, bb, cc]:
answer = 1
return answer
dots.sort(key=lambda x: x[0])
이 코드에서 배울 점
1. is_parallel이라는 함수를 정의하여 코드를 간단히 만듦
2. a, b, c, d = dots로 정의해서 dots 안의 원소를 각각 저장하여 쉽게 꺼내쓸 수 있도록 함
2)
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
3)
import math
from collections import defaultdict
from itertools import permutations
def get_distance(A, B, x, y):
x1, y1 = A
x2, y2 = B
area = abs((x1 - x) * (y2 - y) - (y1 - y) * (x2 - x))
half_line = ((x1 - x2) ** 2 + (y1 - y2) ** 2) ** 0.5
if half_line == 0:
return 0
distance = area / half_line
return distance
def get_slope(A, B):
x1, y1 = A
x2, y2 = B
if y2 - y1 == 0 or x2 - x1 == 0:
return math.inf
return (y2 - y1) / (x2 - x1)
def solution(dots):
for a, b, c, d in permutations(range(4), 4):
slope1 = get_slope(dots[a], dots[b])
slope2 = get_slope(dots[c], dots[d])
if slope1 == slope2:
if get_distance(dots[a], dots[b], *dots[c]) == 0:
continue
if get_distance(dots[a], dots[b], *dots[d]) == 0:
continue
return 1
return 0
'프로그래머스 > 파이썬_입문' 카테고리의 다른 글
💫겹치는 선분의 길이 + 스위핑 (0) | 2025.01.07 |
---|---|
💫유한소수 판별하기 (0) | 2025.01.06 |
저주의 숫자 3 (0) | 2025.01.04 |
외계어 사전 (0) | 2025.01.03 |
💫숨어있는 숫자의 덧셈 (2) (0) | 2024.12.10 |