数据挖掘-knn

最新推荐文章于 2023-12-25 08:16:58 发布

原创最新推荐文章于 2023-12-25 08:16:58 发布 · 197 阅读

0 ·

CC 4.0 BY-SA版权

数据挖掘专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个算法，用于从给定的n个点中找到距离特定点最近的k个点。通过比较每个点到特定点的距离，并维护一个大小为k的候选点集合，该算法能够在遍历所有点后输出这k个点的详细信息。

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
struct node {
	int id;
	double x, y;
}a[N], b[N], t;
int len, k;
int n;
double distance(node x, node y) {
	return sqrt((x.x - y.x) * (x.x - y.x) + (x.y - y.y) * (x.y - y.y));
}
int main() {
	scanf("%d", &n);
	for(int i = 1; i <= n; i++) scanf("%lf %lf", &a[i].x, &a[i].y), a[i].id = i;
	len = 0;
	scanf("%d", &k);
	scanf("%lf %lf", &t.x, &t.y);
	double cnt, minn, tmp;
	int id;
	for(int i = 1; i <= n; i++) {
		if(i <= k) {
			b[i] = a[i];
		} else {
			cnt = distance(t, a[i]);	
			minn = 0;
			for(int j = 1; j <= k; j++) {
				tmp = distance(t, b[j]);
				if(tmp > minn) {
					id = j;
					minn = tmp;
				}
			}
			if(minn > cnt) {
				b[id] = a[i];
			}
		}
	}
	for(int i = 1; i <= k; i++)
		printf("%d %lf %lf\n", b[i].id, b[i].x, b[i].y);
	return 0;
}