题目:27.移除元素

题目

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

不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

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

解题思路

解法一 暴力求解

对于这道题目,由于不需要考虑新数组长度后面多余的元素,就不需要创建新的数组并复制元素过去(数组是连续的存储空间,数组申请的长度不能随便增加或者减少,只能重新申请一片空间);我就首先想到可以遍历全数组并将符合的元素全部挪到数组末尾,同时数组长度--,从而形成一个暴力解法的逻辑删除,以下是我第一次尝试的代码:

int removeElement(int* nums, int numsSize, int val)
{
	for (int i = 0; i < numsSize; ++i)
	{
		if (nums[i] == val)
		{
			for (int j = i + 1; j < numsSize; ++j)
			{
				nums[j - 1] = nums[j];
			}
			numsSize--;
		}
	}
	return numsSize;
}

但是在提交的时候提示测试用例[0,1,2,2,3,0,4,2] 2没有通过,我的输出结果[0,1,2,3,0,4]与答案多出来了一个2。反复检查代码,发现这个2就在第一个2的后面,原来是在numsSize--的同时没有对i进行--,就导致了第一个匹配元素之后的那个元素直接略过了扫描。以下是成功通过的解法:

int removeElement(int* nums, int numsSize, int val)
{
	for (int i = 0; i < numsSize; ++i)
	{
		if (nums[i] == val)
		{
			for (int j = i + 1; j < numsSize; ++j)
			{
				nums[j - 1] = nums[j];
			}
			numsSize--;
            i--;//这里是容易忽略的地方
		}
	}
	return numsSize;
}

由于是嵌套了两层循环,所以这种算法的时间复杂度为O(n²),空间复杂度为O(1)

解法二 快慢指针

快慢指针的思想就是定义两个数组指针,一开始都指向第一个元素,当快指针指向元素不是要清除元素时,快慢指针同时右移并把快指针的值复制给慢指针;当快指针指向元素是清除元素时,快指针直接右移慢指针不动且不复制,这样就通过单层遍历实现略过清除元素,同时也没有改变数组元素的相对位置,以下是我写的代码:

int removeElement(int* nums, int numsSize, int val)
{
	int slow = 0;
	for (int fast = 0; fast < numsSize; fast++)
	{
		if (val != nums[fast])
		{
			nums[slow++] = nums[fast];
		}
	}
	return slow;
}

这种算法的时间复杂度为O(n),空间复杂度为O(1);

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值