一,题目要求
-
给你一个长度固定的整数数组 arr ,请你将该数组中出现的每个零都复写一遍,并将其余的元素向右平移。
-
注意:请不要在超过该数组长度的位置写入元素。请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。
-
示例 1:
输入:arr = [1,0,2,3,0,4,5,0]
输出:[1,0,0,2,3,0,0,4]
解释:调用函数后,输入的数组将被修改为:[1,0,0,2,3,0,0,4]
示例 2:输入:arr = [1,2,3]
输出:[1,2,3]
解释:调用函数后,输入的数组将被修改为:[1,2,3]提示:
1 <= arr.length <= 104
0 <= arr[i] <= 9
-
二,算法原理
-
先找到最后一个复写的数
双指针算法:
定义两个指针cur 和 dest,
cur 指向最后一个被复写的数
dest 指向复写完成之后的最后一个位置。- 判断cur的位置
- 根据cur决定dest向后移动一步或者两步
- 判断一下dest是否已经到结束位置
- cur ++,向后遍历
特殊情况:最后一次复写0时,dest发生数组越界
处理方法:
只将n-1位置处复写0,cur后撤一步,dest后撤两步
- 开始复写
三 , 代码实现
public void duplicateZeros(int[] arr) {
int cur = 0,dest = -1;n = arr.length;
// 找到最后一个需要复写的数
while(cur < n) {
if(arr[cur] == 0) dest += 2;
else dest += 1;
if(dest >= n-1) break;
cur++;
}
// 处理越界情况
if(dest == n) {
arr[n-1] = 0;
cur--;
dest -= 2;
}
// 从后向前完成复写操作
while(cur >= 0) {
if(arr[cur] != 0) arr[dest--] = arr[cur--];
else{
arr[dest--] = 0;
arr[dest--] = 0;
cur--;
}
}
}