381 ut pass

Author: aQua

Algorithms/0381.insert-delete-getrandom-o1-duplicates-allowed/insert-delete-getrandom-o1-duplicates-allowed.go

@@ -26,6 +26,7 @@ func (r *RandomizedCollection) Insert(val int) bool {
 	}
 	r.a = append(r.a, val)
 	r.idx[val] = append(r.idx[val], len(r.a)-1)
+
 	return res
 }
 
@@ -43,7 +44,7 @@ func (r *RandomizedCollection) Remove(val int) bool {
 
 	// 把 a 的最后一个数，放入最后一个 val 的位置
 	r.a[indexOfLastVal] = last
-	r.idx[last][numOfLast-1] = indexOfLastVal
+	r.idx[last] = append([]int{indexOfLastVal}, r.idx[last][:numOfLast-1]...)
 
 	// 删除最后一个数
 	r.a = r.a[:len(r.a)-1]

Algorithms/0381.insert-delete-getrandom-o1-duplicates-allowed/insert-delete-getrandom-o1-duplicates-allowed_test.go

@@ -29,11 +29,11 @@ func Test_Problem0382(t *testing.T) {
 
 		question{
 			para{
-				[]string{"RandomizedCollection", "insert", "insert", "insert", "insert", "insert", "remove", "remove", "remove", "insert", "remove", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom"},
-				[]int{0, 1, 1, 2, 2, 2, 1, 1, 2, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
+				[]string{"RandomizedCollection", "insert", "insert", "insert", "insert", "insert", "remove", "remove", "remove", "insert", "remove", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "getRandom", "remove", "remove"},
+				[]int{0, 1, 1, 2, 2, 2, 1, 1, 2, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1},
 			},
 			ans{
-				[]interface{}{nil, true, false, true, false, false, true, true, true, true, true, 1, 2, 2, 1, 2, 2, 1, 1, 2, 2},
+				[]interface{}{nil, true, false, true, false, false, true, true, true, true, true, 1, 2, 2, 1, 2, 2, 1, 1, 2, 2, true, false},
 			},
 		},
 
@@ -42,6 +42,10 @@ func Test_Problem0382(t *testing.T) {
 
 	for _, q := range qs {
 		sol, ord, par := q.ans.solutions, q.para.orders, q.para.paras
+
+		ast.Equal(len(sol), len(ord))
+		ast.Equal(len(sol), len(par))
+
 		rs := Constructor()
 		for i := 1; i < len(ord); i++ {
 			switch ord[i] {