1. 程式人生 > >LintCode 172. 刪除元素

LintCode 172. 刪除元素

給定 log pos 返回 else post nbsp strong rem

給定一個數組和一個值,在原地刪除與值相同的數字,返回新數組的長度。

元素的順序可以改變,並且對新的數組不會有影響。

樣例

給出一個數組 [0,4,4,0,0,2,4,4],和值 4

返回 4 並且4個元素的新數組為[0,0,0,2]

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 if(A.size()==0) return 0; for(int i=0;i<A.size();) { if(A[i]==elem) { A.erase(A.begin()+i); } else { i++; } }
return A.size(); } };

LintCode 172. 刪除元素