LeetCode 27. Remove Element

题意

给出一个数组和一个数，原地删除数组中与给出数值相等的数，即不能另开数组，求操作完后的数组长度。元素的顺序可以改变，超出数组长度的数不考虑。

思考

最开始的时候想到 vector.erase() ，但是因为未考虑到删除元素后 vector 长度发生改变，所以出错。然后看了别人的代码才意识到错误。

第二个思路是将末尾不等于所给数值的元素移动到前面，但未考虑到数组为空、只有一个数值、如何判断退出循环等边界条件，屡次出错。最后加入了很多条件判断，导致代码看上去很丑陋。

看了别人的代码后，发现思路很清晰，代码很优雅。之后在做题时应尽可能全面的思考问题，找出实现比较简洁的写法。

代码

测试代码：

#include <bits/stdc++.h>
using namespace std;

void print_vector(vector<int>& v) {
    for (auto& i : v) {
        cout << i << " ";
    }
    cout << endl;
}

int main(void) {
    Solution s;
    vector<int> nums{ 0, 1, 2, 2, 3, 0, 4, 2 };
    cout << s.removeElement(nums, 2) << endl;
    print_vector(nums);
    return 0;
}

代码 1：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] == val) {
                nums.erase(nums.begin() + i);
                i--;
            }
        }
        return nums.size();
    }
};

代码 2：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int pos = nums.size() - 1;
        for (int i = 0; i <= pos; i++) {
            if (nums[i] == val) {
                while (pos >= 0 && nums[pos] == val) pos--;
                if (pos < 0 || i > pos) break;
                swap(nums[i], nums[pos--]);
            }
            if (i > pos) break;
        }
        return pos + 1;
    }
};

题解

拷贝覆盖：一个指针指向当前的位置，另一个指针来判断是否与所给数值相等。用后面的值覆盖前面的值。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int pos = 0;
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] != val) {
                nums[pos++] = nums[i];
            }
        }
        return pos;
    }
};

交换移除：一个指针指向当前位置，一个指针指向末尾，如果当前元素与所给数值相同，则与末尾交换。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int pos = 0, ans = nums.size() - 1;
        while (pos <= ans) {
            if (nums[pos] == val) {
                nums[pos] = nums[ans--];
            } else {
                pos++;
            }
        }
        return pos;
    }
};

参考

• 移除元素 - 力扣(LeetCode)
• 画解算法：27. 移除元素 - 画手大鹏