文章目录
- 题目解析
- 算法原理
- 异地
- 原地
- 代码实现
题目解析
题目链接:1089.复写零
这题的暴力解法还是很简单的,不过这道题因为加了两个限制之后,多了一些细节需要去处理。我们通过一个例子来讲解这道题目:
在这个示例中,我们将第一个0都复写了一遍之后,将2和3都右移了一位,然后再复写了一个0,最后将4右移即完成复写,数组长度依然为8,且没有额外开空间,注意这个函数是不需要返回值的。
算法原理
异地
像这种在数组中操作数组元素的这类题目,我们一般也是用双指针算法来解决的,但是我们的双指针算法并不是凭空来的,而是先根据“异地操作”,然后优化成双指针下的“就地”操作。
首先我们先不管题目,再定义一个数组,让cur遍历原数组,遇到非0元素就抄下来,遇到0元素就抄两遍,抄完之后,cur++,dest++,直到cur==n完成复写。
原地
首先在原地的操作中我们要是从前往后复写,则会覆盖,所以我们从后往前复写。(PS:这里我们需要寻找逆向复写的位置以及注意边界问题)
代码实现
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
int cur = 0,dest = -1,n = arr.size();
//寻找逆向复写的位置
while(cur < n)
{
if(arr[cur]) dest++;
else dest += 2;
if(dest >= n-1) break;
cur++;
}
//处理边界问题
if(dest == n)
{
arr[n - 1] = 0;
dest -= 2;
--cur;
}
//从后向前完成复写
while(cur >= 0)
{
if(arr[cur])
{
arr[dest--] = arr[cur--];
}
else
{
arr[dest--] = 0;
arr[dest--] = 0;
--cur;
}
}
}
};