[路飞]_最小的k个数和最接近原点的 K 个点

这是我参与2022首次更文挑战的第5天，活动详情查看：2022首次更文挑战」。
题目地剑指-Offer 40. 最小的k个数
题目地址-973. 最接近原点的 K 个点

题目描述-Offer 40. 最小的k个数

输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。 示例 1：

输入： arr = [3,2,1], k = 2
输出： [1,2] 或者 [2,1]

示例 2：

输入： arr = [0,1,2,1], k = 1
输出： [0]

题目描述-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]] 也会被接受。）

解题思路

• 对输入的数据进行从小到大排序，然后取出前k个即可。
• 将每个点到原点的距离的平方从小到大排序后，取出前 k 个即可

代码1

var getLeastNumbers = function(arr, k){
    let list = [];
    for(let a of arr){
        let flag = false;
        // 实现排序
        for(let j = 0; j < list.length; j++){
            if(list[j] > a){
                list.splice(j, 0, a);
                falg = true;
                break;
            }
        }
        if(!flag){
            list.push(a);
        }
        if(list.lenght > k) list.pop();
    }
    return list;
}

代码1

var kClosest = function(points, k) {
    let list = [];
    for(let a of arr){
        let flag = false;
        // 实现排序
        for(let j = 0; j < list.length; j++){
            // 取出当前值，求平方和
            let aa = a[0] ** 2 + a[1] ** 2;
            let listData = list[j][0] ** 2 + list[j][1] ** 2;
            if(listData > aa){
                list.splice(j, 0, a);
                falg = true;
                break;
            }
        }
        if(!flag){
            list.push(a);
        }
        if(list.lenght > k) list.pop();
    }
    return list;
}