K Closest Points to Origin
Problem Link: https://leetcode.com/problems/k-closest-points-to-origin/
import heapq
class Solution:
def kClosest(self, points: List[List[int]], k: int) -> List[List[int]]:
minheap=[]
heapq.heapify(minheap)
for x,y in points:
dist=(x*x)+(y*y)
heapq.heappush(minheap,[dist,x,y])
ans=[]
while(k>0):
dist,x,y=heapq.heappop(minheap)
ans.append([x,y])
k-=1
return ans
Last updated