//力扣题号:27.移除元素
#include<iostream>
#include<vector>
using namespace std;
//暴力解法:
//时间复杂度为O(n^2),空间复杂度为O(1)
int remove(int target,vector<int>& arry) {
int len = arry.size();//将初始数组的长度赋为len
for (int i = 0; i < arry.size(); i++)//遍历整个数组
{
if (arry[i] == target) {//如果有与目标移除值相等的元素
for (int j = i;j < arry.size()-1;j++) {//从该元素开始,后面的元素依次覆盖前面的元素,注意最后一个需要覆盖的元素为实际数组的倒数第二个元素,并用最后一个元素去覆盖
arry[j] = arry[j + 1];
}
i--;//由于我们将后面的元素覆盖了该元素,后面的元素就跑到前一个去了,一次下标要回溯一下
len--;//我们的数组实际长度-1
}
}
return len;
}
//双指针法:
//时间复杂度为O(n),空间复杂度为O(1)
int removeElement(int target, vector<int>& arry) {
int slowPoint = 0;//定义一个慢指针
//然后再定义一个快指针,这里的指针是指数组元素的下标
for (int fastPoint = 0; fastPoint < arry.size(); fastPoint++)
{
if (target != arry[fastPoint]) {//如果用快指针遍历数组全部元素的时候,元素与目标移除值相等
arry[slowPoint++] = arry[fastPoint];//那么把该元素放进这个数组,相当于慢指针记录的是与目标值不等最终放在数组里面的元素,快指针遍历的是所有元素
//如果有与目标值相等的元素,那么快指针继续向后移动,但是慢指针没有动,慢指针就比快指针慢了一步
//慢的那一步就是空缺出来的位置,后面如果遇到与目标元素不等的元素,则慢指针移动将该元素放进空缺
}
}
return slowPoint;//由于慢指针从0开始计数,按理来说要-1,但最后一个元素记录之后又++了,因此抵消,就是返回本身就行。
}
int main() {
vector<int> arry = { 1, 2, 2, 3, 3, 4 };
cout<<remove(2, arry)<<endl;
cout << removeElement(3, arry);
return 0;
}
运行结果