27. Remove Element

来源:互联网 发布:系统分析师 知乎 编辑:程序博客网 时间:2024/05/29 13:06

题目

给定一个数组和一个值,删除数组中值为给定的值的元素,并返回新数组的长度。

注意:不要申请额外的空间,元素的顺序是可以改变的。并且不关心超出新的数组长度的部分的元素。

Example

给定的输入数组 nums = [ 3,2,2,3 ],val = 3,你的函数应该返回length = 2,并且数组的前两个元素都为2。

解答思路

本题的难点在于不要申请额外的空间,所以,空间复杂度要求是O(1)

本题有两个可以利用的条件。第一个是,本题不关心超出新数组长度部分的元素,第二个是,可以改变原来数组的顺序。

我们完全都不需要使用第二个条件,就逐个遍历数组,如果当前元素等于给定的值,则遍历下一个元素,反之,则保存。

下面则是该思路的Java程序:

public int removeElement(int[] nums, int val) {        if(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;    }

由程序可以看出,该算法的时间复杂度最大为O(n),空间复杂度为O(1)

原创粉丝点击