BOJ / 1004번 / 어린 왕자 [Go][Python3]

👩🏻‍💻 문제

1004번: 어린 왕자

---

✍🏻 풀이

🎨 Go

// https://www.acmicpc.net/problem/1004
package main

import (
	"bufio"
	"fmt"
	"os"
)

func main() {
	reader := bufio.NewReader(os.Stdin)
	writer := bufio.NewWriter(os.Stdout)
	defer writer.Flush()

	var t int
	fmt.Fscanln(reader, &t)

	for i := 0; i < t; i++ {
		var count int
		var startX, startY, endX, endY, n int
		fmt.Fscanln(reader, &startX, &startY, &endX, &endY)
		fmt.Fscanln(reader, &n)
		for j := 0; j < n; j++ {
			var planetX, planetY, r int
			fmt.Fscanln(reader, &planetX, &planetY, &r)
			isStartContained := checkPlanetContains(startX, startY, planetX, planetY, r)
			isEndContained := checkPlanetContains(endX, endY, planetX, planetY, r)
			if isEndContained && !isStartContained || !isEndContained && isStartContained {
				count++
			}
		}
		fmt.Fprintln(writer, count)
	}

}

func checkPlanetContains(x, y, planetX, planetY, r int) bool {
	dist := (x-planetX)*(x-planetX) + (y-planetY)*(y-planetY)
	if dist > r*r {
		return false
	}
	return true
}

🎨 Python3

# https://www.acmicpc.net/problem/1004
import sys

def check_planet_contains(x, y, planet_x, planet_y, r):
    dist = (x-planet_x)**2 + (y-planet_y)**2
    if dist > r*r:
        return False
    return True

if __name__ == "__main__":
    t = int(sys.stdin.readline())
    for i in range(t):
        count = 0
        start_x, start_y, end_x, end_y = list(map(int, sys.stdin.readline().split()))
        n = int(sys.stdin.readline())
        for j in range(n):
            planet_x, planet_y, r = list(map(int, sys.stdin.readline().split()))
            is_start_contained = check_planet_contains(start_x, start_y, planet_x, planet_y, r)
            is_end_contained = check_planet_contains(end_x, end_y, planet_x, planet_y, r)
            if is_end_contained and not is_start_contained or not is_end_contained and is_start_contained:
                count += 1
        print(count)