【双指针算法】--复写零

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.1 先判断cur位置的值
   • 1.1.2 根据cur位置的值判断dest向前走一步还是两步
   • 1.1.3 判断一下dest是否已经到结束位置
   • 1.1.4 cur++
2. 处理边界情况
   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. 从后向前完成复写操作
   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--;
        }
    }
}