백준 1002 터렛 (파이썬)

문제: https://www.acmicpc.net/problem/1002

 

문제

조규현과 백승환은 터렛에 근무하는 직원이다. 하지만 워낙 존재감이 없어서 인구수는 차지하지 않는다. 다음은 조규현과 백승환의 사진이다.

이석원은 조규현과 백승환에게 상대편 마린(류재명)의 위치를 계산하라는 명령을 내렸다. 조규현과 백승환은 각각 자신의 터렛 위치에서 현재 적까지의 거리를 계산했다.

조규현의 좌표 (x1, y1)와 백승환의 좌표 (x2, y2)가 주어지고, 조규현이 계산한 류재명과의 거리 r1과 백승환이 계산한 류재명과의 거리 r2가 주어졌을 때, 류재명이 있을 수 있는 좌표의 수를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스는 다음과 같이 이루어져 있다.

한 줄에 x1, y1, r1, x2, y2, r2가 주어진다. x1, y1, x2, y2는 -10,000보다 크거나 같고, 10,000보다 작거나 같은 정수이고, r1, r2는 10,000보다 작거나 같은 자연수이다.

출력

각 테스트 케이스마다 류재명이 있을 수 있는 위치의 수를 출력한다. 만약 류재명이 있을 수 있는 위치의 개수가 무한대일 경우에는 -1을 출력한다.

 

첫번째 제출 코드 (오답)

import math
import sys
input = sys.stdin.readline
n = int(input().strip())
for i in range(n):
    x1, y1, r1, x2, y2, r2 = map(int, input().split())
    if x1 == x2 and y1 == y2 :
        if r1 == r2:
            print(-1)
        else:
            print(0)
        continue
    d = math.sqrt(pow(abs(x1-x2), 2) + pow(abs(y1-y2), 2))
    #print(d, r1+r2)
    if r1 + r2 > d:
        print(2)
    elif r1 + r2 == d:
        print(1)
    elif r1 + r2 < d:
        print(0)

단순하게 두 원의 중점이 외부에 있는 것만 생각해서 틀렸습니다.

https://www.acmicpc.net/board/view/38854

글 읽기 - ★☆★☆★ 터렛 FAQ ★☆★☆★ 터렛 FAQ ★☆★☆★

이 글의 도움을 얻어 두 원의 위치 관계가 총 6개 있다는 것을 깨닫고 다시 작성하게 되었습니다.

두번째 제출 코드 (정답)

import math
import sys
input = sys.stdin.readline
n = int(input().strip())
for i in range(n):
    x1, y1, r1, x2, y2, r2 = map(int, input().split())
    if x1 == x2 and y1 == y2 :
        if r1 == r2:
            print(-1)
        else:
            print(0)
        continue
    d = math.sqrt(pow(abs(x1-x2), 2) + pow(abs(y1-y2), 2))
    # 한 원이 내부에 있는 경우
    if d < max(r1, r2):
        if abs(r1 - r2) == d:  # 내접
            print(1)
            # 안 내부에 있으나 접하지 않은 경우
        elif d+min(r1, r2) < max(r1, r2):
            print(0)
        else:
            print(2)

    else:
        if r1 + r2 == d:
            print(1)
        elif r1 + r2 < d:
            print(0)
        else:
            print(2)

원의 중점이 다른 원 내부에 포함되는 경우와 그렇지 않은 경우로 나누었습니다. 이를 나누는 기준은 가장 긴 반지름이 두 중점 사이의 거리보다 긴지 아닌지를 체크했습니다.

그 후 접하거나 접하지 않거나 서로 다른 두 점에서 만나는 경우로 나누어주었습니다.