leetcode--remove_element

题意： 给定一个数组和一个值，从数组中删除和该值相同的元素，要求不引入额外的空间，返回新的数组长度，元素的顺序可以调换。
分析： 如果直接删除的话，需要将后面的值往前面移动，这是非常耗费时间的，最坏时间复杂度可以到$O(n^2)$ ，我们添加一个int类型的数count，表示总共有多少个重复的数，初始值为0。在找到一个相同的元素时，直接和A.length - count - 1的调换即可。

代码

public int removeElement(int[] A, int elem) {
        //每次找到一个就把它和最后的调换位置
        int count = 0;
        for(int i = 0; i < A.length - count; i++){
            if(A[i] == elem){
                int temp = A[A.length - count - 1];
                A[A.length - count - 1] = A[i];
                A[i] = temp;
                i--;    //因为最后一个数也有可能是elem，所以这个数也要判断
                count++;    //最后再将count加1
            }
        }
        return A.length - count;
    }

源程序下载