leetCode力扣之Remove Element

Given an array and a value, remove all instances of that value in-place and return the new length.

Do not allocate extra space for another array, you must do this by modifying the input array in-place with O(1) extra memory.

The order of elements can be changed. It doesn't matter what you leave beyond the new length.

Example:

Given nums = [3,2,2,3], val = 3,

Your function should return length = 2, with the first two elements of nums being 2.

/**
 * Created by lxw, liwei4939@126.com on 2018/3/5.
 */
public class RemoveElement {

    public int removeElement(int[] nums, int val){

        if (nums == null || nums.length == 0){
            return 0;
        }

        int len = 0;
        for (int i = 0; i < nums.length; i++){
            if (nums[i] != val){
                nums[len++] = nums[i];
            }
        }
        return len;
    }

    public static void main(String[] args){
        RemoveElement tmp = new RemoveElement();
        int[] nums = new int[]{3, 2, 2, 3, 4};
        int val = 3;
        System.out.print(tmp.removeElement(nums, val));
    }
}