[백준 - python] 1004번: 어린 왕자

문제 링크

1004번: 어린 왕자

 

문제 풀이

t = int(input())

for _ in range(t):
    x1, y1, x2, y2 = map(int, input().split())
    n = int(input())
    cnt = 0

    for _ in range(n):
        cx, cy, r = map(int, input().split())
        r1 = ((x1 - cx) ** 2 + (y1 - cy) ** 2) ** 0.5
        r2 = ((x2 - cx) ** 2 + (y2 - cy) ** 2) ** 0.5
        if (r1 < r and r2 > r) or (r1 > r and r2 < r):
            cnt += 1
    print(cnt)

1002 입출력 예시

*key point: 출발점과 도착점 각각의 행성계의 중심과의 거리와 반지름을 비교하여 출발점과 도착점 중 하나만 행성계 안에 속할 때 진입/이탈 횟수(cnt 변수)에 1을 더해준다.