BOJ-1004 어린 왕자 Swift

문제

어린 왕자는 소혹성 B-664에서 자신이 사랑하는 한 송이 장미를 위해 살아간다. 어느 날 장미가 위험에 빠지게 된 것을 알게 된 어린 왕자는, 장미를 구하기 위해 은하수를 따라 긴 여행을 하기 시작했다. 하지만 어린 왕자의 우주선은 그렇게 좋지 않아서 행성계 간의 이동을 최대한 피해서 여행해야 한다. 아래의 그림은 어린 왕자가 펼쳐본 은하수 지도의 일부이다.

빨간 실선은 어린 왕자가 출발점에서 도착점까지 도달하는데 있어서 필요한 행성계 진입/이탈 횟수를 최소화하는 경로이며, 원은 행성계의 경계를 의미한다. 이러한 경로는 여러 개 존재할 수 있지만 적어도 3번의 행성계 진입/이탈이 필요하다는 것을 알 수 있다.

위와 같은 은하수 지도, 출발점, 도착점이 주어졌을 때 어린 왕자에게 필요한 최소의 행성계 진입/이탈 횟수를 구하는 프로그램을 작성해 보자. 행성계의 경계가 맞닿거나 서로 교차하는 경우는 없다. 또한, 출발점이나 도착점이 행성계 경계에 걸쳐진 경우 역시 입력으로 주어지지 않는다.

입력

입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트케이스에 대해 첫째 줄에 출발점 (x1, y1)과 도착점 (x2, y2)이 주어진다. 두 번째 줄에는 행성계의 개수 n이 주어지며, 세 번째 줄부터 n줄에 걸쳐 행성계의 중점과 반지름 (cx, cy, r)이 주어진다.

출력

각 테스트 케이스에 대해 어린 왕자가 거쳐야 할 최소의 행성계 진입/이탈 횟수를 출력한다.

내가 푼 풀이

- 주어진 행성 안으로 들어가면 진입, 행성밖으로 나가면 이탈 한 총 횟수를 구한다.

- 행성은 중점과 반지름이 주어진다.

- 행성이 안, 밖으로 나가는 경우의수를 구한다.

 

- 출발점과 도착점이 같은 행성에 있다면, 진입/이탈이 필요없음

- 출발점과 도착점이 다른 행성 안에있다면, 진입/이탈 필요함

- 출발점과 도착점중 한지점만 행성안에 있다면 이탈이 필요함

- 출발점과 도착점이 모두 행성밖에 있다면 진입/이탈 필요없음

위 경우의수를 종합해보면 출발점과 도착점이 행성 안에 존재하는지 알아야한다.

 

한지점이 행성안에 존재한다면 지점과 행성의 중점거리는 행성의 반지름보다 작아야한다.

한지점이 행성 밖에 존재한다면 지점과 행성의 중점거리는 행성의 반지름보다 커야한다.

 

이를 바탕으로 코드로 구현해보자

import Foundation

let T = Int(readLine()!)!

for i in 0..<T {
    //입력받기
    let input = readLine()!.split(separator: " ").map{Int(String($0))!}
    let start = [input[0], input[1]]
    let arrive = [input[2], input[3]]
    let N = Int(readLine()!)!
    var count = 0
    
    // 주어진 모든 행성안에 출발점, 도착점이 있는지 확인
    for j in 0..<N {
        // x,y좌표에서 피타고라스의정리를 이용한 두 지점의 거리를 구하는방법 사용
        let planet = readLine()!.split(separator: " ").map{Int(String($0))!}
        let sDist = ((start[0] - planet[0]) * (start[0] - planet[0])) + ((start[1] - planet[1]) * (start[1] - planet[1]))
        let aDist = ((arrive[0] - planet[0]) * (arrive[0] - planet[0])) + ((arrive[1] - planet[1]) * (arrive[1] - planet[1]))
        let powR = planet[2] * planet[2]
        
        // 출발점이 반지름보다 짧고, 도착점이 반지름보다 길다면, 출발점은 행성안에 존재 -> 이탈필요
        if sDist < powR {
            if aDist > powR {
                count += 1
            }
        }
        // 출발점이 반지름보다 길고, 도착점이 반지름보다 짧다면, 도착점은 행성안에 존재 -> 진입필요
        if aDist < powR {
            if sDist > powR {
                count += 1
            }
        }
        
        // 나머지 경우의수들은 모두 행성밖에 존재하거나, 같은행성 안에존재
    }
    print(count)
}