2020-11-09 打卡題-最接近原點(diǎn)的 K 個(gè)點(diǎn)
我們有一個(gè)由平面上的點(diǎn)組成的列表 points括尸。需要從中找出 K 個(gè)距離原點(diǎn) (0, 0) 最近的點(diǎn)。
（這里架曹，平面上兩點(diǎn)之間的距離是歐幾里德距離密浑。）
你可以按任何順序返回答案。除了點(diǎn)坐標(biāo)的順序之外敷钾，答案確保是唯一的。示例 1：
輸入：points = [[1,3],[-2,2]], K = 1
輸出：[[-2,2]]
解釋：
(1, 3) 和原點(diǎn)之間的距離為 sqrt(10)肄梨，
(-2, 2) 和原點(diǎn)之間的距離為 sqrt(8)阻荒，
由于 sqrt(8) < sqrt(10)，(-2, 2) 離原點(diǎn)更近众羡。
我們只需要距離原點(diǎn)最近的 K = 1 個(gè)點(diǎn)侨赡，所以答案就是 [[-2,2]]。示例 2：
輸入：points = [[3,3],[5,-1],[-2,4]], K = 2
輸出：[[3,3],[-2,4]]
（答案 [[-2,4],[3,3]] 也會(huì)被接受。）
提示：
1 <= K <= points.length <= 10000
-10000 < points[i][0] < 10000
-10000 < points[i][1] < 10000來(lái)源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/k-closest-points-to-origin
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有羊壹。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系官方授權(quán)蓖宦，非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。
- 題解：對(duì)二維數(shù)組中的點(diǎn)取橫豎坐標(biāo)平方和作為比較值油猫，進(jìn)行從小到大排序稠茂，取前K個(gè)做返回，涉及Arrays的二維數(shù)組排序規(guī)則重寫情妖，以及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]+"]");
}
}
}
