LeetCode 27. 移除元素

题目描述： 移除元素

        给定一个数组和一个值，在这个数组中原地移除指定值和返回移除后新的数组长度。

        不要为其他数组分配额外空间，你必须使用 O(1) 的额外内存原地修改这个输入数组。

        元素的顺序可以改变。超过返回的新的数组长度以外的数据无论是什么都没关系。

示例:

给定 nums = [3,2,2,3]，val = 3，

你的函数应该返回 长度 = 2，数组的前两个元素是 2。

解题思路：

        遍历数组，若和val相等，则用最后一项代替这一项，并判断代替后的值是不是和val相等。

代码：

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