27. Remove Element

Given an array nums and a value val, remove all instances of that value in-place and return the new length.

Do not allocate extra space for another array, you must do this by modifying the input array in-place with O(1) extra memory.

The order of elements can be changed. It doesn't matter what you leave beyond the new length.

Example 1:

Given nums = [3,2,2,3], val = 3,

Your function should return length = 2, with the first two elements of nums being 2.

It doesn't matter what you leave beyond the returned length.

C++代码：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        vector<int>::iterator it = nums.begin();
        
        for(;it != nums.end();){
            if(*it == val)
                it = nums.erase(it);
            else ++it;
        }
        
        return nums.size();
    }
};

Java代码：

class Solution {
    public int removeElement(int[] nums, int val) {
        int count = 0;
        
        for(int j = 0; j < nums.length; j++){
            if(nums[j]!=val){
                nums[count]=nums[j];
                count++;
            }
        }
        
        return count;
      
    }
}