前言
问题:给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/remove-element
这个问题实现起来并不难,难的是让你使用最优的办法求解,偶尔在b站上看到一个特别妙的解法,满足了文中所给的时间复杂度和空间复杂度的要求,令人拍案叫绝(本人没有学过数据结构,下面只是提供思路,只用一种测试用力,如果考虑不周,自行加上完善,大佬勿cue,狗头)
一、暴力解法,时间复杂度O(n^2)
思路:遍历数组,如果遇到需要删除的数字,使用后面的数字整体移动,进行覆盖
int removeElement(int* nums, int numsSize, int val) {
int i = 0;
int j = 0;
for (i = 0; i < numsSize; i++)
{
if (nums[i] == val)//如果相等,后面的数据往前移动,覆盖掉相等的数
{
for (j = i; j < numsSize - 1; j++)
{
nums[j] = nums[j + 1];
}
numsSize--;
i--;//不能增加,抵消i++作用
}
}
}
int main()
{
int i = 0;
int nums[8] = { 0, 1, 2, 2, 3, 0, 4, 2 };
int val = 2;
int tmp = removeElement(nums, 8, 2);
printf("%d\n", tmp);
for (i = 0; i < tmp; i++)
{
printf("%d ", nums[i]);
}
return 0;
}
二、时间复杂度O(n),空间复杂度O(n)
思路:开辟一个相同大小(足够大)的数组,如果遇到不需要删除的数字,把它复制到我们新开辟的数组中,遍历一遍数组之后,再将新开辟数组中的数据拷贝到原数组中。
int removeElement(int* nums, int numsSize, int val) {
int tmp[8] = { 0 };
int i = 0;
int j = 0;
int newnumsSize = 0;
for (i = 0; i < numsSize; i++)
{
if (nums[i] != val)
{
tmp[j] = nums[i];
j++;
}
}
newnumsSize = j;
for (i = 0; i < newnumsSize; i++)
{
nums[i] = tmp[i];
}
return newnumsSize;
}
int main()
{
int i = 0;
int nums[8] = { 0, 1, 2, 2, 3, 0, 4, 2 };
int val = 2;
int tmp = removeElement(nums, 8, 2);
printf("%d\n", tmp);
for (i = 0; i < tmp; i++)
{
printf("%d ", nums[i]);
}
return 0;
}
三、双指针解法,时间复杂度O(n),空间复杂度O(1)
思路:使用两个下标,src和dst,如果遇到了不需要删除的数字,二者都++,同时把Src的数字复制到dst位置,如果遇到了需要删除的数字,只有src++。
#define _CRT_SECURE_NO_WARNINGS 1
#include <stdio.h>
int removeElement(int* nums, int numsSize, int val) {
int src = 0;
int dst = 0;
while (src < numsSize)
{
if (nums[src] != val)
{
nums[dst] = nums[src];
src++;
dst++;
}
else
{
src++;
}
}
return dst;
}
int main()
{
int i = 0;
int nums[8] = {0, 1, 2, 2, 3, 0, 4, 2};
int val = 2;
int tmp = removeElement(nums, 8, 2);
printf("%d\n", tmp);
for (i = 0; i < tmp; i++)
{
printf("%d ", nums[i]);
}
return 0;
}
运行结果如图所示
总结
双指针解法时间复杂度和空间复杂度都是最优,是一种非常巧妙的思路。