前言
Weekly Contest 141的 复写零:
给你一个长度固定的整数数组
arr
,请你将该数组中出现的每个零都复写一遍,并将其余的元素向右平移。注意:请不要在超过该数组长度的位置写入元素。
要求:请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。
示例1:
输入:[1,0,2,3,0,4,5,0]输出:null解释:调用函数后,输入的数组将被修改为:[1,0,0,2,3,0,0,4]
示例2:
输入:[1,2,3]输出:null解释:调用函数后,输入的数组将被修改为:[1,2,3]
提示:
1 <= arr.length <= 10000
0 <= arr[i] <= 9
解题思路
本题十分简单,需要注意本题是一个原地算法,所以使用双指针法进行变量交换即可。大致逻辑如下:
- 遍历数组元素
- 当前元素为0时,下一个元素赋值为0,并对数组进行左移,下一次遍历当前元素后2位的元素。否则正常遍历数组。
实现代码
/** * 1089. 复写零 * @param arr */ public void duplicateZeros(int[] arr) { for (int i = 0; i < arr.length - 1;) { if (arr[i] == 0) {// 当前元素为0 // 记录下一个元素的值 int next = arr[i + 1]; // 将下一个元素设为0 arr[i + 1] = 0; for (int j = i + 2; j < arr.length; j++) { // 从当前索引后的第二个元素开始交换 int tmp = arr[j]; // 记录当前变量 arr[j] = next; // 将前一个位置的元素值赋给当前元素 next = tmp; // next指针记录当前元素 } i+=2; // 下一次遍历的是当前元素后2位的元素 }else{ ++i; // 正常遍历 } } }