Leetcode每日刷题之 1089. 复写零(C++)

ZZZ_O^O2024-08-14 23:55

1. 题目解析

由题目可知，我们需要将给定数组中的所有零复写一遍，并且保证不能超出数组长度，还不可以开辟新的数组，要在原数组的位置直接操作

2. 算法原理

1. 由题目我们可以将问题简化为找出最终复写完成数组的最后一个元素，然后从该元素由后向前在原数组依次复写遇到的零元素即可

2. 所以我们首先的目标就是找出复写完成后的数组最后的一个元素是什么，那么我们可以使用双指针的方法，即一个为快指针一个为慢指针， cur 从头开始作为慢指针向后遍历， dest 作为快指针由 - 1 位置开始向后遍历，当 cur 指针遇见非零元素则两个指针都向后移一步，如果 cur 指针遇见了零元素，那么就将 cur 元素向后移一步，将 dest 指针向后移两步，直到 dest 指向数组的末尾，那么这时的 cur 指针指向的元素就是复写后的数组的最后一个元素

3. 当然需要注意不可以越，界访问，即当 dest 指针指向了数组末尾的下一个位置，这时就不能直接对 dest 指针指向的区域操作赋值，所以我们可以直接将数组末尾元素赋值为0，然后根据 cur 指针当前指向的元素进行 dest 指针的移动即可

cpp 复制代码

class Solution 
{
public:
    void duplicateZeros(vector<int>& arr) 
    {
        int cur = 0;
        int dest = -1;
        int n = arr.size();
        //1. 找到最终数组的最后一个元素
        while(cur < n)
        {
            if(arr[cur])
            {
                dest++;
            }
            else
            {
                dest += 2;
            }
            if(dest >= n - 1)
            {
                break;
            }
            cur++;
        }

        //2. 处理越界情况
        if(dest == n)
        {
            arr[n - 1] = 0;
            cur--;
            dest -= 2;
        }

        //3.复写零
        while(cur >= 0)
        {
            if(arr[cur])
            {
                arr[dest--] = arr[cur--];
            }
            else
            {
                arr[dest--] = 0;
                arr[dest--] = 0;
                cur--;
            }
        }
    }
};

3. 代码实现