文章目录
- 1. 题目
- 2. 题目解析
- 3. 代码
1. 题目
在线oj
给你一个长度固定的整数数组 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
2. 题目解析
解法:双指针算法
先根据“异地”操作,然后优化成双指针下的就地操作。
步骤:
- 先找到最后一个复写的数
1.1 双指针算法- 1.1.1 先判断cur位置的值
- 1.1.2 根据cur位置的值判断dest向前走一步还是两步
- 1.1.3 判断一下dest是否已经到结束位置
- 1.1.4 cur++
- 处理边界情况
arr[arr.length-2] = 0;
cur–;
dest -= 2;
假设我们现在有这样一个数组,我们按照第一步来找出cur和dest的位置:
cur指向的不是0,dest++, cur++
cur指向的是0,cur++, dest+=2
cur指向的不是0,dest++, cur++
cur指向的不是0,dest++, cur++
cur指向的是0,cur++, dest+=2
此时可以看到dest指向的位置是arr.length,如果不做任何处理就进行第四步的复写操作,那么就会出现数组越界的情况,所以要对dest进行一些处理。
- 从后向前完成复写操作
3.1 如果cur所指向的位置是0,dest的位置和dest-1的位置都要复写为0
3.2 如果cur所指向的位置不是0,那么dest的位置变成cur当前所指向的位置的值
3. 代码
class Solution {public void duplicateZeros(int[] arr) {//1. 找到最后一个要复写的数int cur = 0;int dest = -1;int n = arr.length;for(int i = 0; i < n; i++){if(arr[cur] == 0){dest += 2;}else{dest++;}if(dest>= n-1){break;}cur++;}//2. 处理边界情况if(dest==n){arr[n-1] = 0;cur--;dest-=2;}//3. 从后向前完成复写操作while(cur >= 0){if(arr[cur] == 0){arr[dest] = 0;dest--;arr[dest] = 0;}else{arr[dest] = arr[cur];}dest--;cur--;}}
}