一、题目解析
二、算法原理 (根据”异地“操作优化为”就地“操作)
利用双指针
先正向操作:(会出现元素覆盖)
dest指向0,cur指向0,当cur指向的元素不是0,就cur++,dest++;当cur指向的元素是0,就把dest与dest+1位置变成0,dest++,dest++,cur++;这样的操在异地操作时可行,但就地操作时会覆盖到cur所指向的元素
考虑反向操作:(dest与cur之间有间距,这样可以避免元素的覆盖)
起始:先找到最后一个复写的数,dest指向n-1,cur指向最后一个要复写的数,当cur指向的数不是0,就cur--,dest--;当ur指向的数是0,就dest=0,dest-1=0,dest-=2,cur--;当cur指向-1时,循环结束,这样是可行的
算法设计:
1.先找到最后一个要复写的数:
初始时,cur=0,dest=-1,当cur指向的数不是0,dest++,检查dest是否到达最后一个位置(以下称越界),cur++,;当cur指向的数是0,dest+=2,检查dest是否越界,cur++。当dest>=n-1(指向最后一个位置)跳出循环,cur所指的数就是最后一个要复写的数,dest也刚好指向最后一个位置。
2.cur从后往前遍历,dest从后往前覆盖,当cur指向的数不是0,a[dest]=a[cur],cur--,dest--;当cur指向的数是0,a[dest=0],a[dest-1]=0,dest-=2,cur--;当cur指向-1时,循环结束
3.此外有一种特殊情况,像是[8,4,5,0,0,0,0,7],画图可知,这种情况下dest最后指向n的位置(最后一个元素的下一个位置),导致dest比前面的情况多1,这种情况只会在最后一个复写元素是0时才会出现,我们需要特殊处理这种越界情况:
arr[n-1]=0;cur--;dest-=2
三、代码编写
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
//初始量设置
int dest=-1,cur=0,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;
cur--;
dest-=2;
}
//从后往前复写
while(cur>=0)
{
if(arr[cur])
{
arr[dest--]=arr[cur--];
}
else
{
arr[dest--]=0;
arr[dest--]=0;
cur--;
}
}
}
};