题目
给你一个数组 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);