[백준 / BOJ] 1004번 어린 왕자 (C++, Python)

링크 : https://www.acmicpc.net/problem/1004

1004번: 어린 왕자

---

문제

---

문제 풀이

터렛 문제와 유사한 아이디어를 가진다. 사진과 코드의 주석을 참고하면 된다.

 

1번의 경우에는 dist1 < radius, dist2 > radius 이므로 경유해야 한다. 이는 dist1과 dist2가 바뀌어도 성립한다.

2번의 경우에는 dist1 < radius, dist2 < radius 이므로 경유 자체를 하지 않아도 된다.

 

 

C++ 코드 전문

	#include <algorithm>
	#include <iostream>
	#include <vector>
	void init() {
	std::ios_base::sync_with_stdio(false);
	std::cin.tie(nullptr);
	std::cout.tie(nullptr);
	}
	int main() {
	init();
	int T;
	std::cin >> T;
	for (int t = 0; t < T; ++t) {
	int cnt = 0, n, x1, y1, x2, y2; // 출발점, 도착점
	std::cin >> x1 >> y1 >> x2 >> y2;
	std::cin >> n;
	for (int k = 0; k < n; ++k) {
	int x, y, r;
	std::cin >> x >> y >> r;
	// 출발점과 행성계 거리, 도착점과 행성계 거리
	int dist1 = (x1 - x) * (x1 - x) + (y1 - y) * (y1 - y);
	int dist2 = (x2 - x) * (x2 - x) + (y2 - y) * (y2 - y);
	int radius = r * r;
	// 출발점 - 행성계 거리가 radius보다 크고, 도착점 - 행성계 거리가 radius보다 크면 경유할 의미가 없음. 둘 중 하나만 크면 경유함
	if (dist1 < radius && dist2 < radius)
	continue;
	else if (dist1 < radius || dist2 < radius)
	cnt++;
	}
	std::cout << cnt << "\n";
	}
	return 0;
	}

view raw 1004_어린 왕자 (22.06.05 성공).cpp hosted with ❤ by GitHub

 

Python 코드 전문

	import sys
	input = sys.stdin.readline
	T = int(input())
	for i in range(T):
	cnt = 0
	x1, y1, x2, y2 = map(int, input().split())
	n = int(input())
	for i in range(n):
	x, y, r = map(int, input().split())
	dist1 = (((x1 - x) ** 2) + ((y1 - y) ** 2))
	dist2 = (((x2 - x) ** 2) + ((y2 - y) ** 2))
	radius = r * r
	if dist1 < radius and dist2 < radius:
	continue
	elif dist1 < radius or dist2 < radius:
	cnt += 1
	print(cnt)

view raw 1004_어린 왕자 (22.06.05 성공).cpy hosted with ❤ by GitHub

 

소감