Commit b310a65

committed

K Closest Points to Origin

1 parent c063ec9 commit b310a65Copy full SHA for b310a65

File tree

+20

-0

lines changed

+20

-0

lines changed

Lines changed: 20 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,27 @@`
`29`	`29`	`-104 < xi, yi < 104`
`30`	`30`	`"""`
`31`	`31`	`import math`
	`32`	`+import heap`
`32`	`33`
	`34`	`+`
	`35`	`+# Time Complexity: NLOGK`
	`36`	`+class Solution:`
	`37`	`+ def kClosest(self, points: List[List[int]], k: int) -> List[List[int]]:`
	`38`	`+ heap = [(-self.getDistance(points[index]), index) for index in range(k)]`
	`39`	`+`
	`40`	`+ heapq.heapify(heap)`
	`41`	`+`
	`42`	`+ for index in range(k, len(points)):`
	`43`	`+ heapq.heappushpop(heap, (-self.getDistance(points[index]), index))`
	`44`	`+`
	`45`	`+ return [points[index] for _, index in heap]`
	`46`	`+`
	`47`	`+ def getDistance(self, point):`
	`48`	`+ x, y = point`
	`49`	`+ return math.sqrt((x2) + (y2))`
	`50`	`+`
	`51`	`+`
	`52`	`+# Time Complexity: NLogN`
`33`	`53`	`class Solution:`
`34`	`54`	`def kClosest(self, points: List[List[int]], k: int) -> List[List[int]]:`
`35`	`55`	`distance = []`

Comments

(0)