리트코드 973번 K closest points to origin (Python)

Problem

https://leetcode.com/problems/k-closest-points-to-origin/description/

Solution

from typing import List

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

원점으로부터의 거리를 이용하여 정렬한 뒤 k개 만큼 슬라이싱 하였다.

Reference

파이썬 알고리즘 인터뷰 64번