1
+ //! 给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
2
+ //!
3
+ //! 不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并「原地」修改输入数组。
4
+ //!
5
+ //! 元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
6
+ //!
7
+ //! 示例 1:
8
+ //! 给定 nums = [3,2,2,3], val = 3,
9
+ //! 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
10
+ //! 你不需要考虑数组中超出新长度后面的元素。
11
+ //!
12
+ //! 示例 2:
13
+ //! 给定 nums = [0,1,2,2,3,0,4,2], val = 2,
14
+ //! 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
15
+
16
+
17
+
18
+ /// 时间复杂度
19
+ /// 空间复杂度
20
+ struct Solution ( ) ;
21
+ impl Solution {
22
+
23
+ pub fn remove_element ( nums : & mut Vec < i32 > , val : i32 ) -> i32 {
24
+ let mut slow_index = 0 ;
25
+ for fast_index in 0 ..nums. len ( ) {
26
+ if nums[ fast_index] != val {
27
+ nums[ slow_index] = nums[ fast_index] ;
28
+ slow_index += 1 ;
29
+ }
30
+ }
31
+
32
+ slow_index as i32
33
+ }
34
+ }
35
+
36
+
37
+ #[ cfg( test) ]
38
+ mod tests {
39
+ use super :: * ;
40
+
41
+ #[ test] fn common_test ( ) {
42
+
43
+ let mut v = vec ! [ 3 , 2 , 2 , 3 ] ;
44
+ assert_eq ! ( Solution :: remove_element( & mut v, 3 ) , 2 ) ;
45
+
46
+ assert_eq ! ( v[ 0 ] , 2 ) ;
47
+ assert_eq ! ( v[ 1 ] , 2 ) ;
48
+
49
+
50
+
51
+ let mut v = vec ! [ 0 , 1 , 2 , 2 , 3 , 0 , 4 , 2 ] ;
52
+ assert_eq ! ( Solution :: remove_element( & mut v, 2 ) , 5 ) ;
53
+
54
+ assert_eq ! ( v[ 0 ] , 0 ) ;
55
+ assert_eq ! ( v[ 1 ] , 1 ) ;
56
+ assert_eq ! ( v[ 2 ] , 3 ) ;
57
+ assert_eq ! ( v[ 3 ] , 0 ) ;
58
+ assert_eq ! ( v[ 4 ] , 4 ) ;
59
+ }
60
+
61
+
62
+ }
0 commit comments