lintCode 删除元素

删除元素

给定一个数组和一个值，在原地删除与值相同的数字，返回新数组的长度。
元素的顺序可以改变，并且对新的数组不会有影响。
样例
给出一个数组 [0,4,4,0,0,2,4,4]，和值 4返回 4 并且4个元素的新数组为[0,0,0,2]

思路：这题很简单，保持两个索引 i , j 就好了 用 i 来表示不相同的数字， j 来遍历数组

code :

class Solution {public:    /**      *@param A: A list of integers     *@param elem: An integer     *@return: The new length after remove     */    int removeElement(vector<int> &A, int elem) {        // write your code here        int i = 0;        for (int j = 0; j != A.size(); ++j) {            if(A[j] != elem)                A[i++] = A[j];        }        return i;    }};