这道题是数组数据的删除操作,是一道比较简单的算法题。用双指针就可以实现空间复杂度为O(1)的操作,先看一下题目吧:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 :
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
思路:题目告诉我们这道题要求在原数组中进行删除操作,仅能使用O(1)额外的数组空间。我的第一反应是建一个新数组,并导入原数组中除val以外的其他元素。但O(1)的额外空间让我不能这么操作,只能在原数组上进行元素移动或移除了。
双指针是满足O(1)空间复杂度的操作方式。其逻辑是创建两个指针,使指针A从数组头部开始遍历,如果A指针指向的元素值不为val,则将其值赋给B指针,使B指针同时移动;若A遍历到了值为var的元素,则不保留该元素,A指针继续走,B指针原地不动。这样A指针遍历完一遍后就可以将所有的val元素从数组中剔除掉,最坏情况下数组中没有val元素,B指针需要遍历完整个数组。代码如下:
class Solution {
public int removeElement(int[] nums, int val) {
if(nums == null)return 0;
int len = nums.length;
int j = 0;
for(int i = 0;i<len;i++){
if(nums[i]!=val){
nums[j] = nums[i];
j++;
}
}return j;
}
}
这个方法已经满足题目需求了,空间复杂度为O(1),但最坏情况下需要遍历两遍数组,这一点我们可以从指针的位置上来优化一下。
优化的方法也比较简单,同样是两个指针,指针A从头开始遍历,指针B则从数组尾部开始遍历。当指针A遍历到val元素时,用指针B的元素将其替换掉,当两个指针重合时结束遍历。通过这种方法,在任何情况下我们都只需要遍历一遍数组。
class Solution {
public int removeElement(int[] nums, int val) {
int left = 0;
int right = nums.length;
while (left < right) {
if (nums[left] == val) {
nums[left] = nums[right - 1];
right--;
} else {
left++;
}
}
return left;
}
}