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.

解法

一开始我的思路是：遍历数组，碰到相同的元素，然后将之后的元素前移，所以时间复杂度为$O(n)$，开销大。然后在写代码具体执行中，边界条件一直控制不好。
我的错误代码：

int removeElement(int A[], int n, int elem) {
    int i, j;
    i = 0;
    while (i < n){
        if (n == 1) {
            if (A[0] == elem) {
                A[0] = NULL;
                return 0;
            }
            else
                return 1;
        }
        if (A[i] == elem) {
            for (j = i; j < n - 1; j++) {
                A[j] = A[j + 1];
                --n;
            }
            if (j == n - 1)
                --n;
        }
        else
            ++i;
    }
    return n;
}

可以看到思路很乱，测试用例通不过。而且我一直想着将元素移除的这回事！（比如A[0] = NULL）但其实通过控制数组长度就可以了，应该删除的元素存在分配的内存中就行，只要保证在数组允许访问的长度内访问不到就行。而且题目中只要求返回删除元素后的数组长度。

修改后的代码1：

int removeElement(int A[], int n, int elem) {
    int i, num;
    num = 0;
    for (i = 0; i < n; i++) {
        if (A[i] != elem)
            A[num++] = A[i];
    }
    return num;
}

修改后的代码2：

int removeElement(int A[], int n, int elem) {
    int i = 0;
    while (i < n) {
        if (A[i] == elem) {
            A[i] = A[n - 1];
            n--;
        }
        else
            i++;
    }
    return n;
}