【LeetCode】9.移除元素

最新推荐文章于 2025-07-01 17:23:05 发布

被月亮晒黑_

最新推荐文章于 2025-07-01 17:23:05 发布

阅读量174

点赞数

CC 4.0 BY-SA版权

分类专栏： C++ LeetCode

本文链接：https://blog.youkuaiyun.com/qq_40314507/article/details/85879427

LeetCode 同时被 2 个专栏收录

23 篇文章

订阅专栏

C++

19 篇文章

订阅专栏

本文介绍了解决LeetCode中移除元素问题的两种高效方法。第一种使用迭代器遍历并删除指定值，第二种采用双指针技巧，通过遍历比较实现原地修改。两种方法均达到O(n)时间复杂度和O(1)空间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

0.题目描述

给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

1.解法一：使用迭代器

此种方法简单明了，迭代器遍历，遇到相等的元素便删除，迭代器使用方法代码：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
    vector<int>::const_iterator itc = nums.begin();

	while (itc != nums.end())
	{
		if (*itc == val)
		{
			itc = nums.erase(itc);
		}
		else
		{
			++itc;
		}
		
	} 
        return nums.size();
        
        
    }
};

此种方法时间复杂度O（n）,空间复杂度O（1）。

用时：

3.解法二：双指针

跟上一篇leetcode解法类似，我们定义两个指针i,j。用j遍历数组，当当前数组元素不等于val时，nums[i]等于nums[j]，具体代码如下：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
   	int i = 0;
	for (int j = 0; j < nums.size(); j++)
	{
		if (nums[j] != val)
		{
			nums[i] = nums[j];
			i++;
		}
	}
        return i;
        
        
    }
};

值得一提的是，这种方法，数组的后半部分如何我们不用管，只需要管通过nums[i]赋值过的数组的前半部分，然后输出前半部分的数值，提交。所以数组长度其实并没有变，只是我们return了前面部分数组的长度值，系统任务前面部分的数组就是我们最好所求的删除过指定值的数组。

时间复杂度O（n）,空间复杂度O（1）。

执行情况如图：