knn之构造kd树和最近邻求取c++实现

最新推荐文章于 2024-07-22 14:37:31 发布

潇洒的谢家小帅

最新推荐文章于 2024-07-22 14:37:31 发布

阅读量1.2k

点赞数

文章标签： knn 最近邻

本文链接：https://blog.csdn.net/u012225754/article/details/50208163

版权

这份代码测试样例为

这样，通过中位数来选取根节点(这样的方法其实在一定程度上是有很大问题的，因为根节点的选取方法不同，会导致整棵树的结构不同，这里由于数据的关系，不能构成完全二叉树，所以在对于特殊的样例来说是会出错的，比如说(10,10)这个测试样例，根本无法找到包含他的子节点（区域），所以会导致出错))。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
using namespace std;
struct node{
    pair<int,int>x;
	int dim;
	node*left;
	node*right;
	node*father;
	node(pair<int,int>p=make_pair(0,0),int dim=0,node*left=0,node*right=0,node*father=0)
	:dim(dim),left(left),right(right),father(father)
	{
	 x=p;	
	}
};
bool cmp1(node*a,node* b)
{
	return a->x.first<b->x.first;
}
bool cmp2(node*a,node*b)
{
	return a->x.second<b->x.second;
}
vector<node*>vec;
node* buildtree(vector<node*>temp,int cnt)
{ 

if(temp.size()==0)
return 0;
else if(temp.size()==1)
return temp[0];
else{

    if(cnt==1)
 	sort(temp.begin(),temp.end(),cmp1);
 	else
 	sort(temp.begin(),temp.end(),cmp2);
 	
 	int mid=temp.size()/2;
    vector<node*>p;
    for(int i=0;i<mid;i++)
    {
    	p.push_back(temp[i]);
    }
    vector<node*>q;
    for(int i=mid+1;i<temp.size();i++)
    {
    	q.push_back(temp[i]);
    }
    node*left=buildtree(p,(cnt+1)%2);
    node*right=buildtree(q,(cnt+1)%2);
    node*fat=new node(make_pair(temp[mid]->x.first,temp[mid]->x.second),cnt,left,right,0);
    if(left!=0)
	left->father=fat;
	if(right!=0)
    right->father=fat;
    //cout<<fat->x.first<<" "<<fat->x.second<<endl;
    return fat;
}
}
void traverse(node*root) 
{
	if(root==0)
	{
	}
	else
	{
		cout<<root->x.first<<" "<<root->x.second<<endl;
		traverse(root->left);
		traverse(root->right);
	}
}
node*find_first_belong(node*key,node*root)
{
	node*temp=root; 
	while(true) //遍历找到其归属的叶节点 
	{
		if(temp->left==0&&temp->right==0)
		{
			
			break;
		}
		else
		{
			int