题目链接 复写0
给你一个长度固定的整数数组 arr ,请你将该数组中出现的每个零都复写一遍,并将其余的元素向右平移。
注意:请不要在超过该数组长度的位置写入元素。请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。
问题分析
如果「从前向后」进⾏原地复写操作的话,由于 0 的出现会复写两次,导致没有复写的数「被覆盖掉」。因此我们选择「从后往前」的复写策略。
通过画图一步步分析问题
但是「从后向前」复写的时候,我们需要找到「最后⼀个复写的数」,因此我们的⼤体流程分两步:
i. 先找到最后⼀个复写的数;
ii. 然后从后向前进⾏复写操作
难点:如何找到最后⼀个复写的数
还是使用双指针解决
初始化两个指针 cur = 0 , dest = 0 ;
找到最后⼀个复写的数:
i. 当 cur < n 的时候,⼀直执⾏下⾯循环:
• 判断 cur 位置的元素:
◦ 如果是 0 的话, dest 往后移动两位;否则, dest 往后移动⼀位。
• 判断 dest 时候已经到结束位置,如果结束就终⽌循环;
• 如果没有结束, cur++ ,继续判断。
判断 dest 是否越界到 n 的位置:坑点:如果越界,
执⾏下⾯三步:
- n - 1 位置的值修改成0 ;
- cur 向移动⼀步;
- dest 向前移动两步。
从 cur 位置开始往前遍历原数组,依次还原出复写后的结果数组:
i. 判断 cur 位置的值:
- 如果是 0 : dest 以及 dest - 1 位置修改成 0 , dest -= 2 ;
- 如果⾮零: dest 位置修改成 0 , dest -= 1 ;
ii. cur-- ,复写下⼀个位置。
代码解决
class Solution {
public void duplicateZeros(int[] arr) {
int cur = 0,dest = -1;
//找到最后⼀个复写的数
while(cur<arr.length){
if(arr[cur] != 0)
dest +=1;
else dest += 2;
if(dest >= arr.length-1)
break;
cur++;
}
//处理边界
if(dest == arr.length){
cur--;
arr[dest-1] = 0;
dest -= 2;
}
//从后向前完成复写
while(cur >= 0){
if(arr[cur] != 0){
arr[dest] = arr[cur];
dest--;
}
else {
arr[dest--] = 0;
arr[dest--] = 0;
}
cur--;
}
}
}