Commit b8d4ee3

authored

Update 380.py

1 parent cd769ac commit b8d4ee3Copy full SHA for b8d4ee3

File tree

+30

-0

lines changed

+30

-0

lines changed

Lines changed: 30 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,37 @@ def remove(self, val: int) -> bool:`
`26`	`26`	`def getRandom(self) -> int:`
`27`	`27`	`return random.choice(self.data)`
`28`	`28`
	`29`	`+`
	`30`	`+class RandomizedSet:`
	`31`	`+`
	`32`	`+ def __init__(self):`
	`33`	`+ self.memory = {}`
	`34`	`+ self.values = []`
	`35`	`+`
	`36`	`+`
	`37`	`+ def insert(self, val: int) -> bool:`
	`38`	`+ if val in self.memory:`
	`39`	`+ return False`
	`40`	`+ else:`
	`41`	`+ self.values.append(val)`
	`42`	`+ self.memory[val]=len(self.values)-1`
	`43`	`+ return True`
	`44`	`+`
`29`	`45`
	`46`	`+ def remove(self, val: int) -> bool:`
	`47`	`+ if val not in self.memory:`
	`48`	`+ return False`
	`49`	`+ else:`
	`50`	`+ val_pos = self.memory[val]`
	`51`	`+ self.memory[self.values[-1]] = val_pos`
	`52`	`+ self.values[val_pos], self.values[-1] = self.values[-1], self.values[val_pos]`
	`53`	`+ self.values.pop()`
	`54`	`+ del self.memory[val]`
	`55`	`+ return True`
	`56`	`+`
	`57`	`+`
	`58`	`+ def getRandom(self) -> int:`
	`59`	`+ return random.choice(self.values)`
`30`	`60`
`31`	`61`
`32`	`62`	`# Your RandomizedSet object will be instantiated and called as such:`

Comments

(0)