leetcode 27 -- Remove Element

Remove Element

题目：
Given an array and a value, remove all instances of that value in place and return the new length.
The order of elements can be changed. It doesn’t matter what you leave beyond the new length.

---

题意：
找到数组中的value并且删除，返回数组的新长度。

---

思路：
c++注意删除后迭代器失效即可。

---

代码：

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