双指针(2)_原地复写_复写零问题

C++忠实粉丝

已于 2024-09-04 22:08:44 修改

阅读量1k

点赞数 31

分类专栏：经典算法练习文章标签：数据结构算法 c++

于 2024-09-04 21:45:06 首次发布

本文链接：https://blog.csdn.net/wer24_25/article/details/141903013

版权

经典算法练习专栏收录该内容

49 篇文章 0 订阅

订阅专栏

个人主页：C++忠实粉丝
欢迎点赞👍 收藏✨ 留言✉ 加关注💓本文由 C++忠实粉丝原创

双指针(2)_原地复写_复写零问题

收录于专栏【经典算法练习】
本专栏旨在分享学习C++的一点学习笔记，欢迎大家在评论区交流讨论💌

1.题目链接:

1089. 复写零 - 力扣（LeetCode）

2.题目描述:

给你一个长度固定的整数数组 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

注意题目要求:

请不要在超过该数组长度的位置写入元素。请对输入的数组 就地 进行上述修改，不要从函数返回任何东西。

也就是说:我们不能额外创建一个数组tmp进行存储,只能原地复写

这样的话,我们之前使用双指针从左向右扫描的方法就不能解决这道题了,假如cur指针是用来扫描的,dest指针是用来复习的,而当cur遇到0时,dest会复写两次,有可能会把cur还没扫描的数进行复写,那就会导致cur扫描的下一个数不是原数组的数.

3. 解法（原地复写-双指针）：

算法思路：

如果「从前向后」进⾏原地复写操作的话，由于 0 的出现会复写两次，导致没有复写的数「被覆盖掉」。因此我们选择「从后往前」的复写策略。但是「从后向前」复写的时候，我们需要找到「最后⼀个复写的数」，因此我们的⼤体流程分两步：

i. 先找到最后⼀个复写的数；

ii. 然后从后向前进⾏复写操作。

算法流程:

找到最后一个复写的数:

从后向前完成复写:

特殊边界情况:

a. 初始化两个指针 cur = 0 ,dest = 0;

b. 找到最后⼀个复写的数：

         i. 当cur < n 的时候，⼀直执行下面循环：

                • 判断 cur 位置的元素:

                         ◦ 如果是 0 的话， dest 往后移动两位；

                         ◦ 否则， dest 往后移动⼀位。

                • 判断 dest 是否已经到结束位置，如果结束就终止循环；

                • 如果没有结束， cur++ ，继续判断。

c. 判断 dest 是否越界到 n 的位置：

        i. 如果越界，执行下面三步：

                1. n- 1 位置的值修改成 0 ；

                2. cur 向移动⼀步；

                3. dest 向前移动两步。

d.从 cur 位置开始往前遍历原数组，依次还原出复写后的结果数组：

        i.判断 cur 位置的值：

                1.如果是 0 ： dest 以及 dest - 1 位置修改成 0 ， dest -= 2 ；

                2.如果非零： dest 位置修改成 0 ， dest -= 1 ；

        ii. cur- ，复写下⼀个位置。

代码展示:

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

        //2.处理一下边界情况
        if (dest >= arr.size())
        {
            arr[arr.size() - 1] = 0;
            dest -= 2;
            cur--;
        }
        //3.从后向前完成复写操作
        while (cur >= 0)
        {
            if (arr[cur])
                arr[dest--] = arr[cur--];
            else
            {
                arr[dest--] = arr[cur];
                arr[dest--] = arr[cur--];
            }
        }
    }
};