删除元素（LintCode）

删除元素

给定一个数组和一个值，在原地删除与值相同的数字，返回新数组的长度。

元素的顺序可以改变，并且对新的数组不会有影响。

样例

给出一个数组 [0,4,4,0,0,2,4,4]，和值 4

返回 4 并且4个元素的新数组为[0,0,0,2]

 

思路就是把除删除元素外的元素移到最左边，r是最右端非被删除元素，可以在一开始就找到最右端非被删除元素。用i遍历数组，找到最左端被删除元素，把r所指元素赋到i位置即可。代码里我是先找最左端被删除元素，然后再找最右端非被删除元素。时间复杂度是相同的。

 

public class Solution {
    /** 
     *@param A: A list of integers
     *@param elem: An integer
     *@return: The new length after remove
     */
    public int removeElement(int[] A, int elem) {
        int r = A.length - 1;
        if(r < 0) return 0;
        int i = 0;
        
        while(r > i) {
            if(A[i] == elem) {
                while(A[r] == elem && r > 0) r--;
                if(r > i) {
                    A[i] = A[r];
                    r--;
                }
            }
            i++;
        }
        
        if(A[r] == elem) r--;
        
        return r + 1;
    }
}