复写0
- 这道题的代码模拟了我们作为人用人脑处理的这么个情况。
- 从前往后复写会造成数据覆盖问题,我们不得不从后往前写。即,我们在原数组中先找到要放在最后一个位置的元素【遍历原数组,非0放在原位,0就后移一位{搞个变量cur来模拟}】,然后从后往前遍历,不是0就把原值放在末尾,是0就在该位置和该位置的前一个位置都置0。
- 刷题的经验告诉我们,题目都有临界问题。这道题的临界问题就是:12340。
如果不做额外处理,这道题在遍历原数组时,会识别要放在最后一个位置的元素是0,因为是0,我们会计算cur后移1位,造成cur==n,这种特殊情况,我们直接设置结果数组【即作为输出结果的数组】的末尾为0,而【在原数组中要放在最后一个位置的元素】我们设置为末尾0的前一个元素,用来记录【原数组每一个元素在结果数组的新下标】的cur当然也要做处理,直接前移2位,因为末尾0已经默认位结果数组最后一个数据,我们直接不管末尾0。现在的结果数组的末尾我们就要理解为末尾0的前一个了。 - 废话不多说,上代码。
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
int top = 0;
int cur = -1;
int n = arr.size();
for (top = 0; top < n; top++)
{
if (arr[top] != 0)
cur++;
else
cur += 2;
if (cur >= n - 1)
break;
}
if (cur == n)
{
arr[n - 1] = 0;
top--;
cur -= 2;
}
while (top >= 0)
{
if (arr[top] != 0)
arr[cur--] = arr[top--];
else
{
arr[cur--] = 0;
arr[cur--] = 0;
top--;
}
}
}
};