【leetcode】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.

 

---

题解：用一个len变量记录当前数组的长度。每次遇到元素elem，就把len-1指着的元素换到elem元素所在的地方，然后把len--。要注意的地方就是elem在数组结尾的时候，len减一减到和i相等的时候，就不能再减了，应该结束遍历，返回len的值。

代码如下：

class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        int len = n;
        for(int i = 0;i < len;i ++){
            if(A[i] == elem){
                while(A[len - 1] == elem && len > i)
                    len--;
                A[i] = A[len-1];
                if(len > i)
                    len --;
            }
        }
        /*for(int i= 0;i < len;i++)
            cout <<A[i]<<endl;*/
        return len;
    }
};

 还有一种正序的想法，就是设置一个puthere指针，每次发现和elem不想等的元素，就放到puthere所指向的位置；

java代码如下：

public class Solution {
    public int removeElement(int[] A, int elem) {
        int puthere = 0;
        int n = A.length;
        for(int i = 0;i < A.length;i++){
            if(A[i] != elem){
                A[puthere] = A[i];
                puthere++;
            }
            else {
                n--;
            }
        }
        return n;
    }
}