删除元素_LintCode

来源:互联网 发布:在淘宝上开店货源怎么办 编辑:程序博客网 时间:2024/06/03 20:05

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

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

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

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

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 length = A.length;        if(null==A||A.length<=0){            return 0;        }        for(int i=0;i<length;i++){            if(A[i]==elem){                for(int j=i;j<length-1;j++){                    A[j] = A[j+1];                }                length--;                i--;            }        }        return length;    }}