PGS / 당구 연습 / Python

PGS / 당구 연습

풀면서

✅ 가능한 케이스는 ‘위’, ‘왼’, ‘오’, ‘아래’ 벽에 튕기는 경우로 총 네 가지이다.

✅ 거리를 구할 때 대칭을 이용한다. 예를 들어 위쪽 벽을 맞추는 경우, 타겟공을 위쪽 벽을 기준으로 대칭시킨다. 대칭 시킨 점과 머쓱이가 친 공 사이의 거리제곱을 구하면 그게 곧 답이 된다.

✅ 머쓱이가 친 공이 벽을 맞기 전에 타겟공을 맞추는 상황에 대해 예외 처리를 해줘야 한다.

code ⌨️

import math
def dist(x1, y1, x2, y2):
    return (x1-x2)**2 + (y1-y2)**2

def solution(m, n, startX, startY, balls):
    answer = []
    
    for (x, y) in balls:
        dists = []
        
        # 위
        if not (startX == x and y >= startY):
            y_ = n + (n - y)
            dists.append(dist(startX, startY, x, y_))
        # 아래
        if not (startX == x and y < startY):
            y_ = -y
            dists.append(dist(startX, startY, x, y_))
        # 왼
        if not (startY == y and x <= startX):
            x_ = -x
            dists.append(dist(startX, startY, x_, y))
        # 오
        if not (startY == y and x > startX):
            x_ = m + (m - x)
            dists.append(dist(startX, startY, x_, y))
        answer.append(min(dists))
            
    return answer

풀고 나서👆👀

복잡하게 생각말고 가능한 케이스를 나눠서 생각하면 간단하게 해결할 수 있다.
이때 예외 처리 해줘야 하는 상황을 항상 고려할 것.