Commitd68871c

authored

Merge pull requestneetcode-gh#3259 from saurabh10041998/insert-delete-get-random

Create: 0380-insert-delete-getrandom.rs

2 parentsbec4e98 +d482ed5 commitd68871cCopy full SHA for d68871c

File tree

+40

-0

lines changed

+40

-0

lines changed

Lines changed: 40 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,40 @@`
	`1`	`+use rand::seq::SliceRandom;`
	`2`	`+use std::collections::HashMap;`
	`3`	`+`
	`4`	`+pubstructRandomizedSet{`
	`5`	`+mp:HashMap<i32,i32>,`
	`6`	`+arr:Vec<i32>,`
	`7`	`+}`
	`8`	`+`
	`9`	`+implRandomizedSet{`
	`10`	`+fnnew() ->Self{`
	`11`	`+RandomizedSet{`
	`12`	`+mp:HashMap::new(),`
	`13`	`+arr:Vec::new(),`
	`14`	`+}`
	`15`	`+}`
	`16`	`+`
	`17`	`+fninsert(&mutself,val:i32) ->bool{`
	`18`	`+let res = !self.mp.contains_key(&val);`
	`19`	`+if res{`
	`20`	`+self.mp.insert(val,self.arr.len()asi32);`
	`21`	`+self.arr.push(val);`
	`22`	`+}`
	`23`	`+ res`
	`24`	`+}`
	`25`	`+fnremove(&mutself,val:i32) ->bool{`
	`26`	`+let res =self.mp.contains_key(&val);`
	`27`	`+if res{`
	`28`	`+let idx =*self.mp.get(&val).unwrap();`
	`29`	`+self.mp`
	`30`	`+.entry(*self.arr.last().unwrap())`
	`31`	`+.and_modify(\|v\|*v = idx);`
	`32`	`+self.arr.swap_remove(idxasusize);`
	`33`	`+self.mp.remove(&val);`
	`34`	`+}`
	`35`	`+ res`
	`36`	`+}`
	`37`	`+fnget_random(&self) ->i32{`
	`38`	`+*self.arr.choose(&mut rand::thread_rng()).unwrap()`
	`39`	`+}`
	`40`	`+}`

Comments

(0)