【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 = ;
     for (int i = ; i < nums.size(); i++) {
         if (nums[i] == nums[i - ])
             count++;
         else
             nums[i - count] = nums[i];
     }
     return nums.size() - count;
 }

这个更加易懂

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

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

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

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

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

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

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

使用类似的方法很快解决

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