题干:
https://leetcode.cn/problems/remove-element/
- 双指针
和本专栏中的“ 26 删除有序数组中的重复项”很相似,我们采用双指针解决问题。
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
//if (nums.empty()) { throw "the array is empty!"; } //测试中有空数组哈哈哈,很尴尬不符合答案
if (nums.empty()) { return 0; }
int fir{ 0 };//第一个指针,指向被覆盖的空间
int sec{ 0 };//第二个指针,遍历数组
while (sec != nums.size()) {
if (nums[sec] == val) { }
else {
nums[fir] = nums[sec];
fir++;
}
sec++;//因为sec++在if-else中都需要递增,所以写到了这里
}
return fir;//fir代表返回数组的size(),fir-1是返回数组的最大下标
}
};
优化后的代码,赋值操作变少了。很值得一看(代码写得越复杂,逻辑问题可能就越多,像这样简洁的代码才是我们追求的)
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
if (nums.empty()) { return 0; }
//注意区间的开闭
int left{ 0 };//left左边区间 [0,left) 全不等于val
int right{ static_cast<int>(nums.size()-1) };//right右边区间(right,size)全等于val
while (left <= right) {
if (nums[left] == val) {
nums[left] = nums[right];
right--;//只有确保了right被赋值为val,right才左移
}
else {
left++;//只有left不等于val,left才右移
}
}
return left;//或者right+1
}
};