leetcode1089.复写零
题目链接:1089.复写零
题目描述:
给你一个长度固定的整数数组 arr
,请你将该数组中出现的每个零都复写一遍,并将其余的元素向右平移。
注意:请不要在超过该数组长度的位置写入元素。请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。
示例一:
输入: 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
算法原理:
我们可以定义双指针 dest 和 cur 来对数组进行模拟。在草稿纸上先写一个数组,根据异地模拟操作,然后优化成在给定数组下,用双指针就地操作。
- 1.先找到最后一个要“复写”的数
- 先判断 cur 位置的值
- 根据 cur 的值来决定 dest 向后一步或者两步
- 每次都判断 dest 是否达到结束位置(n-1坐标或者n坐标),如果达到位置直接退出循环
- 执行完上边操作,在循环末尾 cur++
- 注意:在 dest 到达 n 坐标的时候,dest 已经越界。(原因就是:(1)这时最后一个复写的数是0,(2)复写0的时候数组只剩一个位置,无法复写两个0,只能复写一个0。)此时,dest 到结束位置时,dest 和 n 的值相等。所以从后向前复写时,只需要让
arr[n - 1] = 0; dest -= 2; cur--;
即可 。 - 2.从后向前完成复写操作
注意:定义的 dest 和 cur 都是数组坐标。,n 位数组元素个数。
代码如下:
class Solution {
public:void duplicateZeros(vector<int>& arr) {//1.“双指针 dest cur”从前向后找到复写后数组最后一个元素 //特例:边界是0的情况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;dest-=2; cur--;}//2.用“双指针”从后向前开始复写while(cur>=0){if (arr[cur]) arr[dest--] = arr[cur--];else{arr[dest--] = 0;arr[dest--] = 0;cur--;}}}
};