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.

看题看了半天，英文不太好啊，大致意思就是，删掉数组中和给定值相同的所有元素，并返回新的数组长度，新数组不要求和原数组顺序一样。

---

别人的代码：

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

 

我的代码。。。。  先排序，然后分三种情况，要找的数在中间，在左边，或者在右边，确实够笨，没想到删除一个元素，直接左移覆盖就行了。

class Solution {
public:
    void swap(int* A,int a,int b)
    {
        int temp=A[a];
        A[a]=A[b];
        A[b]=temp;
    }
    int removeElement(int* A, int n, int elem) {
        int res=0;
        if(A==NULL) 
            return 0;
        sort(A,A+n);//排序
        for(int i=0;i<n;++i)
        {
            if(A[i]==elem)
                ++res;
        }
        if(res==n) 
            return 0;
        if(res==0) 
            return n;
       
　　　　 if(A[0]!=elem&&A[n-1]!=elem)
        {
            int t=1;
            int tag=0;
            for(int i=0;i<n;++i)
            {
                if(res>=t&&A[i]==elem&&A[n-t]!=elem){
                    tag=i;swap(A,i,n-t);++t;
                }
            }
            return n-res;
        }
        else if(A[0]==elem&&A[n-1]!=elem)
        {
            int t=1;
            for(int i=0;i<n;++i)
            {
                if(res>=t&&A[i]==elem&&A[n-t]!=elem){
                    swap(A,i,n-t);++t;
                }
            }
            return n-res;
        }
        else
        {
            for(int i=0;i<n;++i)
            {
                if(A[i]==elem) return n-res;
            }
        }
        return n-res;
    }
};