如果「从前向后」进⾏原地复写操作的话,由于 0 的出现会复写两次,导致没有复写的数「被覆 盖掉」。因此我们选择「从后往前」的复写。 但是「从后向前」复写的时候,我们需要找到「最后⼀个复写的数」,因此我们的⼤体流程分两 步: 先找到最后⼀个复写的数; 然后从后向前进⾏复写操作。
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
// 1找到最后一个要复写的数
int cur = 0, dest = -1;
while (1)
{
if (arr[cur])
dest++;
else
dest += 2;
if (dest >= arr.size() - 1)
break;
cur++;
}
if (dest == arr.size())
{
arr[dest-1] = arr[cur];
dest -= 2;
cur--;
}
// 2 从右往左完成复写操作
while (cur>=0)
{
if (arr[cur])
{
arr[dest--] = arr[cur--];
}
else
{
arr[dest] = arr[dest-1] = 0;
dest -= 2;
cur--;
}
}
}
};