0%

vector容器删除元素的几种简单方法

一、直接删除法

(1)使用 vector库函数“erase”删除,使用erase函数后容器size自动-1
1
2
3
4
5
6
7
8
9
10
int removeElement(vector<int>& nums, int detarget) {
for (int i = 0; i < nums.size(); i++){
if(nums[i] == detarget){
nums.erase(nums.begin() + i);
i--; //由于容器size-1,还按原来的i的话相当于自动右移一位而漏掉一个元素
}
}
return nums.size();
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#include <iostream>
#include <vector>
using namespace std;
int removeElement(vector<int>& nums, int detarget) {
for (int i = 0; i < nums.size(); i++){
if(nums[i] == detarget){
nums.erase(nums.begin() + i);
i--; //由于容器size-1,还按原`` 来的i的话相当于自动右移一位而漏掉一个元素
}
}
return nums.size();
}

int main() {
vector<int> nums ;
nums.push_back(3);
nums.push_back(2);
nums.push_back(2);
nums.push_back(3);
int detarget = 3;
int len = removeElement(nums, detarget);
cout << "After removing " << detarget << ", the length of nums is " << len << endl;
for(int i = 0; i < len; i++){
cout << nums[i] << " ";
}
cout << endl;
return 0;
}
(2)使用 vector库函数“swap和pop_back()”,由于pop_back删除的是最后一个元素,所以先移位再删除
1
2
3
4
5
6
7
8
9
10
int removeElement(vector<int>& nums, int detarget) {
for (int i = 0; i < nums.size(); i++){
if(nums[i] == detarget){
swap(nums[i], nums[nums.size() - 1]); //将要删除的元素交换到最后
nums.pop_back();
i--; //这里的i--与上面的作用一样
}
}
return nums.size();
}

二、遍历覆盖法

如果从数组的角度理解,因为数组存储的内容在地址上是连续的,要移除目标元素,就没有库函数可以使用,那就需要对其他元素进行处理,进行覆盖和前移,比较暴力的思路是发现目标元素后,将后面的所有元素都前移覆盖原来的元素,这种方法需要两个for循环,时间复杂度为O(n2),这里不详细讲这种写法。
从覆盖的角度思考,我们可以通过一次遍历把非目标元素全部前移覆盖到前半部片

1
2
3
4
5
6
int removeElement(vector<int>& nums, int detarget) {
int size = 0;
for(auto x : nums) //遍历全部元素
if(x != detarget) nums[size++] = x;//如果不是要删除的目标元素,把它前移覆盖到前半部分
return size; //这里的size即为删除后的容量大小
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
//暴力
int removeElement(int* nums, int numsSize, int val){
int j=0;
int nums2[100];
int i;
for( 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;
}

三、相向双指针法

(1)使用vector库函数swap,通过有两个指针和一个for循环将所有元素分为两个部分,

左端的指针(快指针)控制前半部分的边界,swap后前半部分全部是要保留的元素
右端的指针(慢指针)控制后半部分的边界,swap后后半部分全部是要删除的元素

//时间复杂度:O(n)
//空间复杂度:O(1)

1
2
3
4
5
6
7
8
9
10
11
12


int removeElement(vector<int>& nums, int detarget) {
int j = nums.size() - 1;
for (int i = 0; i <= j; i++) {
if (nums[i] == detarget) {
swap(nums[i--], nums[j--]); //注意这里的i--,j--,代表先交换再移动,而不是先移动再交换
//i--的原因:如果发生交换的两个元素全是要删除的目标元素,这样不会把换过来的漏删
}
}
return j + 1; //j+1就是删除完的size大小
}

ps:上面几种方法只给出了核心算法的代码,具体问题做出相应改动即可

-------------本文结束感谢您的阅读-------------
老板你好,讨口饭吃