Remove Element

public class Solution {
    /**
     *@param A: A list of integers
     *@param elem: An integer
     *@return: The new length after remove
     */
    public static int removeElement(int[] aA, int elem) {
        if (aA == null || aA.length == 0){
            return 0;
        }
        int count = 0;
        for (int i=0; i < aA.length; i++){
            if (elem != aA[i]){
                aA[count++] = aA[i];
            }
        }
        return count;
    }
}