Remove Element

2017/11/30

Given an array and a value, 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.

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