973. 最接近原点的 K 个点 (Python 实现)

题目：

给定一个数组 points ，其中 points[i] = [xi, yi] 表示 X-Y 平面上的一个点，并且是一个整数 k ，返回离原点 (0,0) 最近的 k 个点。
这里，平面上两点之间的距离是 欧几里德距离（ √(x1 - x2)2 + (y1 - y2)2 ）。
你可以按 任何顺序 返回答案。除了点坐标的顺序之外，答案 确保 是 唯一 的。

示例 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]] 也会被接受。）

代码：

class Solution:
    def kClosest(self, points: List[List[int]], k: int) -> List[List[int]]:
        temp1, temp2 = [], []
        for x in points:
            r = x[0]**2 + x[1]**2
            if len(temp1)<k:
                temp1.append(r)
                temp2.append(x)
            else:
                if r < max(temp1):
                    site = temp1.index(max(temp1))
                    temp1[site], temp2[site] = r, x
        return temp2