K Closest Points to Origin

Code

O(NlogN) : 644 ms, faster than 98.60% of Python3 online submissions

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