Commit b8c7c0e

committed

O(nlogn) time and O(n) space using max heap.

1 parent dc8a236 commit b8c7c0eCopy full SHA for b8c7c0e

File tree

-9

lines changed

-9

lines changed

Lines changed: 5 additions & 9 deletions

Original file line number	Diff line number	Diff line change
`@@ -31,12 +31,8 @@ def lastStoneWeight(self, stones: List[int]) -> int:`
`31`	`31`	`max_heap = [-val for val in stones]`
`32`	`32`	`heapq.heapify(max_heap)`
`33`	`33`	`while len(max_heap) > 1:`
`34`		`- y = (-1) * heapq.heappop(max_heap)`
`35`		`- x = (-1) * heapq.heappop(max_heap)`
`36`		`- if x == y:`
`37`		`- if not max_heap: # consider a case [1,1]`
`38`		`- return 0`
`39`		`- continue`
`40`		`- else:`
`41`		`- heapq.heappush(max_heap,x-y)`
`42`		`- return max_heap[0] if max_heap[0] > 0 else (-1)*max_heap[0]`
	`34`	`+ stone1 = (-1) * heapq.heappop(max_heap)`
	`35`	`+ stone2 = (-1) * heapq.heappop(max_heap)`
	`36`	`+ if stone1 != stone2:`
	`37`	`+ heapq.heappush(max_heap,-(stone1-stone2))`
	`38`	`+ return -max_heap[0] if max_heap else 0`

Comments

(0)