如何删除数组中的某个元素?
例:给你一个数组 nums
和一个值 val
,你需要移除所有数值等于 val
的元素,并返回移除后数组的新长度。
三种方法
1.元素前移(时间复杂度:O(N^2),空间复杂度:O(1))
遍历数组,当遇到需要删除的元素时,则将其后面的元素全部前移一个空间,覆盖当前需要删除的元素。前移后下标不变,需要再次判断当前位置的元素是否为要删除元素。
int removeElement(int* nums, int numsSize, int val)
{
int newnumsSize = numsSize;
for (int i = 0; i < numsSize; i++)
{
if (nums[i] == val)
{
newnumsSize--;
for (int j = i; j < numsSize - 1; j++)
nums[j] = nums[j + 1];
i--;//后一元素前移后该元素仍要判断是否为val
}
}
return newnumsSize;
}
int main()
{
int nums[] = { 3,2,6,6,6,6,7,8,9 };
int newnumsSize = removeElement(nums, 9, 6);
for (int i = 0; i < newnumsSize; i++)
printf("%d ", nums[i]);
return 0;
}
2.快慢指针(额外数组,空间换时间)(时间复杂度:O(N),空间复杂度O(N))
创建一个快指针、一个慢指针和一个数组。通过快指针遍历原数组,只要遇到的元素不等于val,那么就将该元素通过慢指针赋值到新数组中。这样快指针遍历完成后,新数组中的元素就是不包含val的元素,再将新数组的元素赋值到原数组中,即可完成删除val。
int removeElement(int* nums, int numsSize, int val)
{
int* newnums = (int*)malloc(sizeof(int) * numsSize);//建立新数组
int fast = 0;
int slow = 0;
for (fast = 0; fast < numsSize; fast++)
{
if (nums[fast] != val)
{
newnums[slow] = nums[fast];
slow++;
}
}
for (int i = 0; i < slow; i++)//将新数组赋值到原数组中
{
nums[i] = newnums[i];
}
return slow;
}
int main()//测试
{
int nums[] = { 3,6,6,6,6,7,8,9,10,11 };
int newnumsSize = removeElement(nums, 10, 6);
for (int i = 0; i < newnumsSize; i++)
{
printf("%d ", nums[i]);
}
return 0;
}
3.快慢指针(非额外数组)(时间复杂度:O(N),空间复杂度O(1))
创建一个快指针和一个慢指针,不再创建新的数组,而是直接在原数组中进行元素的赋值拷贝。(本质其实和额外数组相同,只是将不等于val的元素拷贝到原数组中,这样val元素就会被覆盖或者在数组新长度的界外)通过快指针遍历数组, 如果快指针遍历遇到不等于 val 的元素,则将该元素赋值到慢指针指向的位置,且慢指针位置后移。这样遍历完成后,数组中的val元素就会被“删除”
int removeElement(int* nums, int numsSize, int val)
{
int fast = 0;
int slow = 0;
for (fast = 0; fast < numsSize; fast++)
{
if (nums[fast] != val)
{
nums[slow] = nums[fast];
slow++;
}
}
return slow;
}