2020-11-09 打卡題-最接近原點的 K 個點
我們有一個由平面上的點組成的列表 points。需要從中找出 K 個距離原點 (0, 0) 最近的點。
（這里，平面上兩點之間的距離是歐幾里德距離。）
你可以按任何順序返回答案。除了點坐標的順序之外，答案確保是唯一的。示例 1：
輸入：points = [[1,3],[-2,2]], K = 1
輸出：[[-2,2]]
解釋：
(1, 3) 和原點之間的距離為 sqrt(10)，
(-2, 2) 和原點之間的距離為 sqrt(8)，
由于 sqrt(8) < sqrt(10)，(-2, 2) 離原點更近。
我們只需要距離原點最近的 K = 1 個點，所以答案就是 [[-2,2]]。示例 2：
輸入：points = [[3,3],[5,-1],[-2,4]], K = 2
輸出：[[3,3],[-2,4]]
（答案 [[-2,4],[3,3]] 也會被接受。）
提示：
1 <= K <= points.length <= 10000
-10000 < points[i][0] < 10000
-10000 < points[i][1] < 10000來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/k-closest-points-to-origin
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。
- 題解：對二維數組中的點取橫豎坐標平方和作為比較值，進行從小到大排序，取前K個做返回，涉及Arrays的二維數組排序規則重寫，以及Arrays.copyOfRange方法
import java.util.Arrays;
import java.util.Comparator;
public class KClosest {
public int[][] kClosest(int[][] points, int K) {
Arrays.sort(points, new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
return ((o1[0]*o1[0]+o1[1]*o1[1]) - (o2[0]*o2[0]+o2[1]*o2[1]));
}
});
return Arrays.copyOfRange(points,0, K);
}
public static void main(String args[]){
int points[][] = new int[][]{{1,3},{-2,2}};
int points2[][] = new int[][]{{3,3},{5,-1},{-2,4}};
KClosest kc = new KClosest();
int res[][] = kc.kClosest(points, 1);
int res2[][] = kc.kClosest(points2, 2);
for (int i = 0; i < res.length; i++) {
System.out.println("["+res[i][0]+","+res[i][1]+"]");
}
for (int i = 0; i < res2.length; i++) {
System.out.println("["+res2[i][0]+","+res2[i][1]+"]");
}
}
}
