[LeetCode973. 最接近原点的 K 个点] | 刷题打卡

109 阅读1分钟

一、题目描述:

leetcode-cn.com/problems/ma…
我们有一个由平面上的点组成的列表 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]]

二、思路分析:

  • 直接暴力 ,sort + map 方法有点笨但是还是ac

三、AC 代码:

/**
 * @param {number[][]} points
 * @param {number} k
 * @return {number[][]}
 */
var kClosest = function(points, k) {
    const res = []
    let min = Number.MAX_SAFE_INTEGER
    const map = {}
    for(let i=0;i<points.length;i++){
        const [x,y] = points[i]
        const dis = Math.pow(x,2) + Math.pow(y,2)
        if(map[dis]){
            map[dis].push(points[i])
        }else{
            map[dis] = [points[i]]
        }
    }
    const keys = []
    for(let key in map){
        keys.push(key)
    }
    keys.sort((a,b)=>a-b)
    for(let i=0;i<keys.length;i++){
        for(let j=0;j<map[keys[i]].length;j++){
            if(k ===0){
                return res
            }
            res.push(map[keys[i]][j])
            k--
        }
    }
    return res
};