思路:
1、這個題應該是很簡單的,但是我的代碼寫的比較複雜。
2、如果熟悉優先隊列的朋友應該會寫出比較好的算法。
3、我這個就是先存前K個元素,之後再去找出前K個元素最大一個結果,去和當前的對比。
4、我這個算法之後每次都要便利K次以便找到最大的值。
5、優先隊列先前就排序好了前K個元素。比較的時候就簡單了
class Solution {
class point{
private int x;
private int y;
private double result;
point(int x,int y,double result){
this.x = x;
this.y = y;
this.result = result;
}
}
public int[][] kClosest(int[][] points, int K) {
List<point> list = new ArrayList<>();
for(int i = 0;i < points.length; i++) {
double result = Math.sqrt( (points[i][0]*points[i][0]) + (points[i][1]*points[i][1]) );
if(list.size() < K) {
list.add(new point(points[i][0],points[i][1],result));
} else {
point max = list.get(0);
for(point p : list) {
if(p.result > max.result) {
max = p;
}
}
if(max.result > result) {
max.x = points[i][0];
max.y = points[i][1];
max.result = result;
}
}
}
int[][] returnResult = new int[K][2];
int i = 0;
for(point p : list) {
returnResult[i][0] = p.x;
returnResult[i][1] = p.y;
i++;
}
return returnResult;
}
}