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

x33g5p2x  于2022-02-15 转载在 Python  
字(0.6k)|赞(0)|评价(0)|浏览(132)

题目:
给定一个数组 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

相关文章