취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 17797번 문제인 Dome Construction이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/17797

주어진 좌표들과 원점 사이의 거리들 중 $k$번째로 가까운 점을 포함하는 dome은 다른 그보다 더 가까운(또는 같은 거리인) $k-1$개의 점 또한 포함함을 관찰하자. 따라서 $k$번째로 가까운 점의 거리를 구하는 것으로 문제를 해결할 수 있다.

피타고라스 정리와 정렬을 이용해 문제를 간단하게 해결하자.

아래는 제출한 소스코드이다.

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long double ld;

int N, K;
ld x, y, z;
ld A[100000];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin >> N >> K;
	for (int i = 0; i < N; i++) {
		cin >> x >> y >> z;
		A[i] = x * x + y * y + z * z;
	}
	sort(A, A + N);
	cout << fixed;
	cout.precision(10);
	cout << sqrtl(A[K - 1]);
}