一、直接删除法 (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--; } } 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--; } } 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--; } } 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; }
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--]); } } return j + 1 ; }
ps:上面几种方法只给出了核心算法的代码,具体问题做出相应改动即可