LeetCode27. Remove Element

来源:互联网 发布:dos攻击软件 编辑:程序博客网 时间:2024/06/10 00:47

27. 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 in place with constant memory.

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

Example:
Given input array nums = [3,2,2,3], val = 3

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

题意:给定一个数组和一个值,移除数组中等于这个值的元素,并返回移除后的长度。
例如:数组为[3,2,2,3],值为3,那么返回 length=2,并且数组的前两个元素是2.
要求:不分配另一个数组的空间,数组元素的顺序可以改变,且最终数组的长度无所谓。

思路:本题思路与RemoveDuplicates_26的思路一致,用 i 来标记不等于 val 值的元素个数,用 j 来遍历数组。这种方法最终数组的长度保持不变。

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;}

另一种思路:每次用数组最后一个元素覆盖要移除的元素,并将数组长度减1,这种方法使最终数组长度变小。

public int removeElement1(int[] nums, int val) {    int i = 0;    int n = nums.length;    while (i < n) {        if (nums[i] == val) {            nums[i] = nums[n - 1];            // reduce array size by one             n--;        } else {            i++;        }    }    return n;}