380 runtime error

Author: aQua

Algorithms/0380.insert-delete-getrandom-o1/insert-delete-getrandom-o1.go

@@ -1,59 +1,56 @@
 package Problem0380
 
 import "math/rand"
+import "fmt"
 
 // RandomizedSet 是一个随机获取的集合
 type RandomizedSet struct {
-	rec map[int]bool
+	a   []int
+	idx map[int]int // idx 的 v 是 k 在 a 中的位置。
 }
 
 // Constructor returns RandomizedSet
 /** Initialize your data structure here. */
 func Constructor() RandomizedSet {
 	return RandomizedSet{
-		rec: make(map[int]bool),
+		a:   []int{},
+		idx: make(map[int]int),
 	}
 }
 
 // Insert 插入数据
 /** Inserts a value to the set. Returns true if the set did not already contain the specified element. */
 func (r *RandomizedSet) Insert(val int) bool {
-	if _, ok := r.rec[val]; ok {
+	if _, ok := r.idx[val]; ok {
 		return false
 	}
-	r.rec[val] = true
+	r.a = append(r.a, val)
+	r.idx[val] = len(r.a) - 1
 	return true
 }
 
 // Remove 删除数据
 /** Removes a value from the set. Returns true if the set contained the specified element. */
 func (r *RandomizedSet) Remove(val int) bool {
-	if _, ok := r.rec[val]; ok {
-		delete(r.rec, val)
-		return true
+	if _, ok := r.idx[val]; !ok {
+		return false
 	}
 
-	return false
+	// 把 a 的最后一个数，放入待删除的数的位置
+	r.a[r.idx[val]] = r.a[len(r.a)-1]
+	// 删除最后一个数
+	r.a = r.a[:len(r.a)-1]
+	// 在 r.m 中删除 val 的记录
+	delete(r.idx, val)
+
+	return true
 }
 
 // GetRandom 获取随机数据
 /** Get a random element from the set. */
 func (r *RandomizedSet) GetRandom() int {
-	if len(r.rec) == 0 {
-		return 0
-	}
-
-	rnd := rand.Int() % len(r.rec)
-	res := 0
-	for k := range r.rec {
-		if rnd == 0 {
-			res = k
-			break
-		}
-		rnd--
-	}
-
-	return res
+	fmt.Println(r.a)
+	return r.a[rand.Intn(len(r.a))]
 }
 
 /**

Algorithms/0380.insert-delete-getrandom-o1/insert-delete-getrandom-o1_test.go

@@ -11,8 +11,6 @@ func Test_Problem0380(t *testing.T) {
 
 	rs := Constructor()
 
-	ast.Equal(0, rs.GetRandom(), "从空集合中，只能取出0")
-
 	ast.True(rs.Insert(1), "插入1")
 
 	ast.False(rs.Remove(2), "删除2")