LeetCode – Remove Element (Java)

这题和上题思路完全一致, 双指针. 有了前一题的段练, 这题明显顺多了, 我在leecode oj上直接写的, 第一次在某行漏了个";", 然后第二次就顺利accpeted了. 开心

Code:

public class Solution {
    public int removeElement(int[] A, int elem) {
        if(A.length == 0){
            return 0;
        }
        if(A.length == 1){
            if(A[0] == elem){
                return 0;
            }else{
                return 1;
            }
        }
        int i = 0;
        int j = 0;
        while (j < A.length){
            if (A[j] == elem){
                j++;
            }else{
                A[i] = A[j];
                i++;
                j++;
            }
        }
        return i;
    }
}