973. K Closest Points to Origin - https://leetcode.com/problems/k-closest-points-to-origin/

KClosestPoints.py

class Solution:
    def kClosest(self, points: List[List[int]], K: int) -> List[List[int]]:
        
        """
        in: List, K
        out List
        Time: NLogN due to sorting
        Space: N due to dictionary
        
        """
        
        if not points:
            return []
        
        resd = {}
        res = []
        
        for point in points:
            resd[tuple(point)] = point[0]*point[0] + point[1]*point[1]
            
        resd = sorted(resd.items(), key=lambda x: x[1])
        
    
        for i in range(0, K, 1):
            res.append(list(resd[i][0]))
                
        return res