Use la func ‘dist’ del modulo math para sacar las distancias
import heapq
from math import dist
from typing import List
def kClosest(puntos: List[List[int]], K: int) -> List[List[int]]:
min_heap = []
for p in puntos:
distance = -(dist((0,0), p))
if len(min_heap) < K:
heapq.heappush(min_heap, (distance, p))
else:
if distance > min_heap[0][0]:
heapq.heapreplace(min_heap, (distance, p))
return [i[1] for i in min_heap]
¿Quieres ver más aportes, preguntas y respuestas de la comunidad?
o inicia sesión.