本算法思想来自于李航的《统计学习方法》本文主要实现其kd树最近邻的搜索
构造代码已经发表过了,现在也还是写一下,
//为了方便储存数据
public class Data {
public double x1;
public double x2;
}
//kd树的代码
public class Tree {
public Tree left;//左节点
public Tree father;//父节点
public Tree right;//右节点
public Data mData;//节点的数据
public int split;//判断维数
public void setSplit(int split) {
this.split = split;
}
public int getSplit() {
return split;
}
}
//构造kd树
private static void builtTree(Tree root, Data[] datas) {
if (datas == null) return;//当没有的时候说明这个不能成为节点
if (datas.length == 1) { //只有一个就是他自己了,它和第一个判断不能换位置
root.mData = datas[0];//设置数据
return;
} else {
dataSort(datas, root.getSplit() % 2); //进行数据的排列,注意传参判断其是第几维划分
root.left = new Tree();root.left.father = root;
root.left.setSplit(root.getSplit() + 1);//不要忘记写