题目
分析 && 代码
和前一道LeetCode 26. Remove Duplicates from Sorted Array一样的思路,不过它的follow-up还是有点意思。按照26题的思路,可以写出下面的代码:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int len = 0;
for (int i = 0; i < nums.size(); ++i) {
if (nums[i] != val) nums[len++] = nums[i];
}
return len;
}
};
follow-up是考虑当待删除的元素很少的时候:
不难发现,这个算法对nums所有的元素都检查了一遍,如果对于一个10w大小的nums,要删除1,而且只在最后一个位置是1其他都不是,那这个代码要要复制10w - 1次,i和len加起来变化了20w-1次。
就这一点点小地方,可以做一定优化。
我们可以让指针双向的走,单向,单向要走20w步,而双向可以只走一半,可以这么做的原因是题目说明元素顺序可以变化,来看代码就清楚了:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int i = 0, n = nums.size();
while (i < n) {
if (nums[i] == val) {
nums[i] = nums[n - 1];
n--;
}
else i++;
}
return n;
}
};
这样遍历结束i和n加起来走了n步,就这么一点点优化,两份代码交上去时间没有区别…当作开拓思维吧