LintCode 172. 刪除元素
阿新 • • 發佈:2018-01-28
給定 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, intelem) { // 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. 刪除元素