【LeetCode】数组移除元素

链表等复杂数据结构用多了，简单的数组操作也不能遗忘！

1. 给定一个有序数组，移除所有重复元素并返回新的数组长度，不能分配额外数组的内存空间。

e.g. 给定输入的数组 = [1,1,2]，函数应当返回新长度 = 2，前 2 个元素为 1 和 2。超出新长度所剩下什么元素都无所谓，即新数组可以 = [1,2,2]。

一开始没读懂 “It doesn't matter what you leave beyond the new length.” 的含义，用了 erase 方法非常复杂，答案比较简略。

int removeDuplicates(vector<int>& nums) {
    int count = 0;
    for (int i = 1; i < nums.size(); i++) {
        if (nums[i] == nums[i - 1])
            count++;
        else
            nums[i - count] = nums[i];
    }
    return nums.size() - count;
}

这个更加易懂

int removeDuplicates(vector<int>& nums) {
    if (nums.empty())   return 0;
    int count = 1;
    for (int i = 1; i < nums.size(); i++)
        if (nums[i] != nums[i - 1])
            nums[count++] = nums[i];
    return count;
}

但这样的代码不够“好看”，可以改进为

int removeDuplicates(vector<int>& nums) {
    int i = 0;
    for (int n : nums)
        if (!i || n != nums[i-1])
            nums[i++] = n;
    return i;
}

如果不想用 !i 来做第一个元素的判断，代码还能写的更巧妙

int removeDuplicates(vector<int>& nums) {
    int i = !nums.empty();
    for (int n : nums)
        if (n != nums[i-1])
            nums[i++] = n;
    return i;
}

int i = !nums.empty(); 这样的写法真的很有趣很有趣，值得学习！

2. 给定一个数组和一个值，移除数组中所有等于该值的元素，并返回新数组的长度，不能分配额外数组的内存空间。超出新长度所剩余的元素也不考虑。

使用类似的方法很快解决

int removeElement(vector<int>& nums, int val) {
    int i = 0;
    for(int n : nums) {
        if (n != val)
            nums[i++] = n;
    }
    return i;
}