Skip to content

Commit 991f4cf

Browse files
committed
q27
1 parent 21d5c20 commit 991f4cf

File tree

2 files changed

+63
-0
lines changed

2 files changed

+63
-0
lines changed

src/array/mod.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
11
mod q35;
2+
mod q27;

src/array/q27.rs

Lines changed: 62 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,62 @@
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

Comments
 (0)