自己老师布置的一个题目:在100W个100维的向量中,取出与给定的100维的向量,欧几里得距离最小的前K个向量。
我把这个问题分为了2部分,一个是计算部分,一个是统计部分。统计部分用到了索引堆,很好用!但是在计算部分,我使用最普通的计算公式,太耗费时间(8s)。所以有没有关于欧几里得距离更高效的计算方式?感谢波波老师了。
long start = System.currentTimeMillis();
//100W行向量,这里用数组表示一个向量
for (int i = 1; i < Const.MATRIX_ROW; i++) {
//每一行与给定向量的欧几里得距离
double temp = 0.0;
//具体计算(x-x1)^2+(y-y1)^2+(z-z1)^2+...+
for (int j = 0; j < Const.MATRIX_COL; j++) {
temp+= Math.pow(arr[j]-context.data[i][j],2);
}
//每一行的结果保存在一个数组中
result[i] = temp;
}
long end = System.currentTimeMillis();
System.out.println("普通方法"+(end-start)+"ms");