LintCode Delete Element 删除元素

给定一个数组和一个值，在原地删除与值相同的数字，返回新数组的长度。
元素的顺序可以改变，并且对新的数组不会有影响。
Given an array and a value, remove all occurrences of that value in place and return the new length.
The order of elements can be changed, and the elements after the new length don’t matter.

样例
给出一个数组 [0,4,4,0,0,2,4,4]，和值 4
返回 4 并且4个元素的新数组为[0,0,0,2]
Example
Given an array [0,4,4,0,0,2,4,4], value=4
return 4 and front four elements of the array is [0,0,0,2]

public class Solution {
    /** 
     *@param A: A list of integers
     *@param elem: An integer
     *@return: The new length after remove
     */
     //两根指针遍历数组，一根（i）从前往后，一根(j)从后往前
    public int removeElement(int[] A, int elem) {
        int count = A.length;
        for(int i = 0, j = A.length - 1; i < A.length && i < j; i++) {
            if(A[i] == elem) {//前指针遇到需删除的元素
                count --;
                while(i < j ) {//找到后指针所指向的第一个遇到的不该删除的元素，并赋给A[i]
                    if(A[j] != elem) {
                        A[i] = A[j];
                        j--;
                        break;
                    }else {
                        count--;
                        j--;
                    }
                }
            }
        }
        return count;
    }
}