遇到删除序列中所有不符合条件的元素时,均可以使用以下两种算法,原地修改序列,时间复杂度O(N),空间复杂度O(1)。
其核心思想是把数组分割为两个部分:左侧有效元素序列,右侧无效元素序列,每次找到有效元素就插入到有效元素序列末尾。
如:删除序列中大于等于s,小于等于t的所有元素。
k记录不用删除元素的个数
void deleteTarget(vector<int>arr,int s,int t){
int k=0; /*记录不删除、需要保留下来的元素的个数*/
for(int i=0;i<arr.size();i++){
if(arr[i]<s || arr[i]>t)
arr[k++]=arr[i];
}
arr.resize(k);
}
k记录需要删除元素的个数
void deleteTarget2(vector<int>arr,int s,int t){
int k=0; /*记录需要删除的元素的个数*/
for(int i=0;i<arr.size();i++)
if(arr[i]>=s && arr[i]<=t)
k++;
else
arr[i-k]=arr[i]; /*前面有k个无效元素,故往前挪k个位置*/
}