1、最接近原点的 K 个点（Weekly Contest 119）

题目描述：
我们有一个由平面上的点组成的列表 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

public int[][] kClosest(int[][] points, int K) {
        if(K >= points.length){
        	return points;
        }
        int result [][] = new int[K][2];
        Map<Integer, Integer> tem = new HashMap<Integer, Integer>();
        //将坐标和其对应的值放入map中去
        for (int i = 0; i < points.length; i++) {
			tem.put(i, points[i][0] * points[i][0] + points[i][1] * points[i][1] );
		}
        for (int i = 0; i < result.length; i++) {
			int min = Integer.MAX_VALUE;
			int index = 0;
			for(Map.Entry<Integer, Integer> entry:tem.entrySet()){
				if(entry.getValue() < min){
					index = entry.getKey();
					min = entry.getValue();
				}
			}
			result[i][0] = points[index][0];
			result[i][1] = points[index][1];
			tem.remove(index);
		}
        
        
        return result;
    }

排名靠前的代码

import java.util.concurrent.ThreadLocalRandom;

class Solution {
    int[][] points;
    public int[][] kClosest(int[][] points, int K) {
        this.points = points;
        work(0, points.length - 1, K);
        return Arrays.copyOfRange(points, 0, K);
    }

    public void work(int i, int j, int K) {
        if (i >= j) return;
        int oi = i, oj = j;
        int pivot = dist(ThreadLocalRandom.current().nextInt(i, j));

        while (i < j) {
            while (i < j && dist(i) < pivot) i++;
            while (i < j && dist(j) > pivot) j--;
            swap(i, j);
        }

        if (K <= i - oi + 1)
            work(oi, i, K);
        else
            work(i+1, oj, K - (i - oi + 1));
    }

    public int dist(int i) {
        return points[i][0] * points[i][0] + points[i][1] * points[i][1];
    }

    public void swap(int i, int j) {
        int t0 = points[i][0], t1 = points[i][1];
        points[i][0] = points[j][0];
        points[i][1] = points[j][1];
        points[j][0] = t0;
        points[j][1] = t1;
    }
}