最接近原点的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]] 也会被接受

class Solution:
    def kClosest(self, points, K):
        """
        :type points: List[List[int]]
        :type K: int
        :rtype: List[List[int]]
        """
        return sorted(points, key = lambda x: x[0]*x[0] + x[1]*x[1])[:K]