27. 移除元素

给定一个数组nums和一个值val，你需要原地移除所有数值等于val的元素，返回移除后数组的新长度。不要使用额外的数组空间，你必须在原地修改输入数组并在使用O(1) 额外空间的条件下完成。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

解法

1
2
3
4
5
6
7
8
9
10

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

题解

因为题目说要原地移除，那一定是原地遍历一遍直接赋值。采用快慢指针方式，慢指针指向修正后数组索引，快指针进行遍历。这个题关键点是要找到如果遍历值和val相等则继续遍历，直到不相等进行赋值操作。