leetcode: Remove Element

来源:互联网 发布:成都湛源软件 编辑:程序博客网 时间:2024/05/21 09:19

设置一个末尾指针,如果有遇到相同元素,那么就把相同元素和末尾指针的数对换。

注意因为换过来的数又要重新比较,所以要i--。

public class Solution {

    public int removeElement(int[] A, int elem) {
        int n=A.length;
        if(n==0)
            return 0;
        int i=0,p=n-1;
        for(i=0;i<=p;i++)
        {
            if(A[i]==elem)
            {
                A[i]=A[p];
                p--;i--;
            }
        }
        return p+1;
    }
}
0 0
原创粉丝点击