lintcode：Remove Element 删除元素

题目：

删除元素

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

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

 样例

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

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

解题：

Java程序：

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) {
        // write your code here
        int count = A.length;
        // for(int i=0;i<count;i++)
        int i=0;
        while(i<count){
            if(A[i]==elem){
                
                for(int j=i;j<count-1;j++)
                    A[j] = A[j+1];
                count--;
            }else
                i++;
        }
        return count;
    }
}

View Code