leetcode-27-数组题-移除元素

发表于 | 分类于

题目

1.png

解法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
class Solution {
public:
    // 双指针
    int removeElement(vector<int>& nums, int val) {
        int n = nums.size();
        if(n == 0)
            return 0;

        int i = 0;
        for(int j = 0; j < n; ++j){
            if(nums[j] != val){
                nums[i++] = nums[j];
            }
        }

        return i;
    }
    
    // 双指针-移除元素很少时，避免重复对不需要删除的元素赋值
    int removeElement(vector<int>& nums, int val) {
        int n = nums.size();
        if(n == 0)
            return 0;

        int i = 0;
        while(i < n){
            if(nums[i] == val){
                nums[i] = nums[n-1];
                --n;
            }else{
                ++i;
            }
        }

        return n;
    }
};