leetcode第一刷_Remove Element

水题。

这道题的题干直接把思路说出来了啊。。惊呆了。

两个指针，一前一后，遇到要去掉的数字就把后面的换上来，呵呵。

inline void swap(int &a, int &b){
    int tp = a;
    a = b;
    b = tp;
}

class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        if(A == 0)
            return 0;
        int end = n-1;
        while(A[end--] == elem);
        for(int i=0;i<end;i++){
            if(A[i] == elem){
                swap(A[i], A[end]);
                while(A[--end] == elem);
            }
        }
        return end+1;
    }
};