题目详情:. - 力扣(LeetCode)
一 . 题目解析
题意为:从左到右原地将数组中的0元素复写一遍,且数组的大小不能改变
二 . 算法原理
首先如果我们选择从前向后复写,那么前面复写的元素则会覆盖后面需要复写的数,这样行不通。那么我们就需要从后向前复写来解决前面的问题。
从后向前复写有两个问题:1. 我们应该从后面的什么位置开始复写? 我们不可能从数组的最后一个元素开始复写,这样是不合理的。所以我们应该找到最后一个复写元素。 2.从后向前我们应该如何复写?利用规律找到复写的步骤。
具体流程:
1.找到最后一个复写元素:
定义出两个变量:int cur=0; int dest=-1; cur用于遍历数组,dest用于标记当前复写到了什么位置。当cur指向元素为非0元素则dest++,为0元素则dest+=2。当dest>=数组的最大下标时循环结束,cur指向的位置就是最后复写元素
2.复写步骤
如果dest刚好在数组最后一个元素。那么从cur位置开始从后向前遍历,cur指向元素为非0元素时 数组[dest]=数组[cur] 然后dest--,cur--。若为0元素这需要覆盖两次 数组[dest--]=0; 数组[dest--]=0; 然后cur--。
如果dest在数组范围外,则需要特殊处理一下。如果不处理则会导致越界访问。处理:首先我们要知道dest为什么会跑出数组范围外?那肯定是因为此时cur指向的元素要复写两次,如果此元素只需要复写一次的话根本不会超出范围。那么是什么元素会复写两次?那肯定是0,所以我们需要做的处理就是:将数组最后一个元素修改为0,然后再将cur--,dest-=2,手动的帮它完成复写工作。然后接下来的操作与上面一模一样。
三 . 代码实现
class Solution
{
public:
void duplicateZeros(vector<int>& arr)
{
// 1. 先找到最后⼀个数
int cur = 0, dest = -1, n = arr.size();
while (cur < n)
{
if (arr[cur]) dest++;
else dest += 2;
if (dest >= n - 1) break;
cur++;
}
// 2. 处理⼀下边界情况
if (dest >= n)
{
arr[n-1] = 0;
cur--; dest -= 2;
}
// 3. 从后向前完成复写操作
while (cur >= 0)
{
if (arr[cur]) arr[dest--] = arr[cur--];
else
{
arr[dest--] = 0;
arr[dest--] = 0;
cur--;
}
}
}
};