零、描述
题目链接:27. 移除元素
一、想法:
首先想用两个指针,左边指针从头到尾,右边从尾到头。
当左边指针指向arr[ i ] != target 时候,记录下索引 L,
右指针开始从尾到头,当arr[ j ]==target时候,记录下索引 R 。
int removeElement(vector<int>& nums, int val) {
int N=nums.size();
int l=0,r=N-1; //存储下一次循环的地点
if(N==0) return {};
for(int i=l;i<r;i++){
if(nums[i]==val){
l=i+1; //留作下次开始
for(int j=r;j>=i;j--){
if(nums[j]!=val){
r=j-1;
nums[i]=nums[j];
break;
}
}
}
}
return l+1;
}
当输入nums=[1] val=1 时候,正确结果是[ ] ,但是代码返回的结果是 [1]
二、参考
/**
* 相向双指针方法,基于元素顺序可以改变的题目描述改变了元素相对位置,确保了移动最少元素
* 时间复杂度:O(n)
* 空间复杂度:O(1)
*/
int removeElement(vector<int>& nums, int val) {
int leftIndex = 0;
int rightIndex = nums.size() - 1;
while(leftIndex<=rightIndex){
// 找左边等于val的元素
while (leftIndex <= rightIndex && nums[leftIndex] != val){
++leftIndex;
}
//跳出while时候,leftIndex表示此时左边等于val的元素
// 找右边不等于val的元素
while (leftIndex <= rightIndex && nums[rightIndex] == val) {
-- rightIndex;
}
//跳出while时候,leftIndex表示此时右边不等于val的元素
// 将右边不等于val的元素覆盖左边等于val的元素
if (leftIndex < rightIndex) {
nums[leftIndex++] = nums[rightIndex--];
}
}
return leftIndex; // leftIndex一定指向了最终数组末尾的下一个元素
}
};
举例
nums = [3,2,2,3], val = 3
leftIndex=0, rightIndex=3
nums[ leftIndex=0 ] == val, leftIndex=0
nums[ rightIndex=3 ] == val, rightIndex=2
nums[ rightIndex=2 ] != val 跳出while
到底if ,此时leftIndex=0, rightIndex=2 交换得到nums = [2,2,3,3]
总结
- 采用 while 替代 for
- while实现是也是 if 中符合条件停止左指针 的功能