[알고리즘] 원점에 K번째로 가가까운 점

원점에 K번째로 가가까운 점

내 풀이

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

정렬기준을 lamda를 통해서 넣어주면 된다. x는 element이다. 어짜피 유클리드 거리를 구하는데 sqrt는 있으나마나여서 뺐다.