【算法——双指针】LeetCode 1089 复写零

最新推荐文章于 2025-11-25 14:21:24 发布

原创最新推荐文章于 2025-11-25 14:21:24 发布 · 481 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #数据结构 #算法

刷题专栏收录该内容

20 篇文章

订阅专栏

文章讲述了如何在给定的整数数组中，不改变元素顺序但确保每两个连续的0都被视为一个，通过正序遍历并巧妙地利用dest指针实现。作者提供了一个名为Solution的类方法实现这一功能。

千万不要被这道题标注着“简单”迷惑了，实际上需要注意的细节很多。

题目描述：

解题思路：

正序遍历，确定结果数组的最后一个元素所在的位置；知道最后一个元素的位置后倒序进行填充。

先找到最后一个需要复写的数
1. 先判断cur位置的值
2. 决定dest向后移动一步还是两步
3. 判断dest是否已经到结束位置
4. cur++
处理边界情况（dest超过arr.size()-1的情况）
从后向前完成复写的操作

代码：

class Solution {
public:
    void duplicateZeros(vector<int>& arr) {
        int cur = 0;
        int dest = -1;
        while(cur < arr.size())
        {
            if(arr[cur] == 0)
            {
                dest += 2;
            }
            else
            {
                dest++;
            }
            if(dest >= arr.size() - 1)
            {
                break;
            }
            cur++;
        }
        if(dest > arr.size() - 1)
        {
            arr[arr.size() - 1] = 0;
            cur--;
            dest -= 2;
        }
        while(cur >= 0)
        {
            if(arr[cur] != 0)
            {
                arr[dest] = arr[cur];
                dest--;
                cur--;
            }
            else
            {
                arr[dest] = 0;
                arr[dest - 1] = 0;
                cur--;
                dest -= 2;
            }
        }
    }
};