[前端]_一起刷leetcode 973. 最接近原点的 K 个点

大家好，我是挨打的阿木木，爱好算法的前端摸鱼老。最近会频繁给大家分享我刷算法题过程中的思路和心得。如果你也是想提高逼格的摸鱼老，欢迎关注我，一起学习。

题目

973. 最接近原点的 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. 1 <= K <= points.length <= 10000
2. -10000 < points[i][0] < 10000
3. -10000 < points[i][1] < 10000

思路

1. 我们只需要找到前面k个元素，对于它的顺序并不关心，那么我们可以通过快速排序，每一轮找到index个比当前元素小的数；
2. 如果index的值大于k - 1， 说明我们找到的太多了，那么我们再通过快排从里面删除掉多余的一部分即可；
3. 如果index的值小于k - 1， 说明我们找到的还不够，那么我们再通过快排从里面找出更多即可；
4. 当index的值刚好等于k - 1的时候，我们复制前面k个元素即可。

实现

/**
 * @param {number[][]} points
 * @param {number} k
 * @return {number[][]}
 */
var kClosest = function(points, k) {
    let left = 0, right = points.length - 1;
    let index = quickSort(points, left, right);

    // 只排序到我们想要的区间
    while (index !== k - 1) {
        if (index < k - 1) {
            left = index + 1;  
        } else {
            right = index - 1;
        }
        index = quickSort(points, left, right);
    }
    
    return points.slice(0, k);
};

// 快速排序
function quickSort(arr, left, right) {
    let index = left;
    let baseValue = getMaxPath(arr[left]);

    for (let i = left + 1; i <= right; i++) {
        if (getMaxPath(arr[i]) < baseValue) {
            index++;
            swap(arr, index, i);
        }
    }

    swap(arr, index, left);

    return index;
}

// 勾股定律，直接比就好了没必要开平方
function getMaxPath(arr) {
    return arr[0] * arr[0] + arr[1] * arr[1];
}

// 交换节点的值
function swap(arr, i, j) {
    const temp = arr[i];
    arr[i] = arr[j];
    arr[j] = temp;
}

结果

看懂了的小伙伴可以点个关注、咱们下道题目见。如无意外以后文章都会以这种形式，有好的建议欢迎评论区留言。