题目描述:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
学习记录:
和上题一样,只是判断条件改变。看了题解方法一双指针,思路一样,但是突然发现result和index是同步变化的呀,可以删去一个,虽然设想的用法不同,但实际是一样的。
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int n=nums.size();
if(n==0) return 0;
//int result=0;//记录有多少个不同
int index=0;//记录下一个存储位置
for(int i=0;i<n;i++)
{
if(nums[i]!=val)
{
//result++;
nums[index]=nums[i];
index++;
}
}
//return result;
return index;
}
};
题解:
方法二,双指针优化,由于不需考虑前面元素的位置,所以不需要顺序遍历。然后上述方法是可以的,但是存在很多不需要的操作。比如 [4,5,6] val=4;我们把5放在位置1上,又把6放在位置2上,但不考虑顺序,最简操作显然是将6放在位置1上。所以可以采用双指针的优化,定义左右指针,左指针存放下一个要存的地方,右指针找下一个要存的元素,左右重合就是结束。
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int left = 0, right = nums.size();
while (left < right) {
if (nums[left] == val) {
nums[left] = nums[right - 1];
right--;
} else {
left++;
}
}
return left;
}
};
作者:LeetCode-Solution
链接:https://leetcode.cn/problems/remove-element/solution/yi-chu-yuan-su-by-leetcode-solution-svxi/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。