381 runtime error

Author: aQua

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

@@ -37,19 +37,21 @@ func (r *RandomizedCollection) Remove(val int) bool {
 	}
 
 	last := r.a[len(r.a)-1]
-	lenOfLast := len(r.idx[last])
-	lenOfVal := len(r.idx[val])
+	numOfLast := len(r.idx[last])
+	numOfVal := len(r.idx[val])
+	indexOfLastVal := r.idx[val][numOfVal-1]
+
 	// 把 a 的最后一个数，放入最后一个 val 的位置
-	r.a[r.idx[val][lenOfVal-1]] = last
-	r.idx[last][lenOfLast-1] = r.idx[val][lenOfVal-1]
+	r.a[indexOfLastVal] = last
+	r.idx[last][numOfLast-1] = indexOfLastVal
 
 	// 删除最后一个数
 	r.a = r.a[:len(r.a)-1]
 	// 在 r.idx 中删除最后一个 val 的记录
-	if lenOfVal == 1 {
+	if numOfVal == 1 {
 		delete(r.idx, val)
 	} else {
-		r.idx[val] = r.idx[val][:lenOfVal-1]
+		r.idx[val] = r.idx[val][:numOfVal-1]
 	}
 
 	return true