每日OJ题_算法_双指针②_力扣1089. 复写零

目录

力扣1089. 复写零

解析代码

力扣1089. 复写零

1089. 复写零 - 力扣（LeetCode）

难度 简单

给你一个长度固定的整数数组?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 <= 10^4
• 0 <= arr[i] <= 9

class Solution {
public:
    void duplicateZeros(vector<int>& arr) {

    }
};

解析代码

从右往前的双指针问题（标的简单实际并不简单），需要先找到最后得到的vector最右边的数。这里找这个数用从左向右的双指针：

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

        if(left == size) // 处理边界情况
        {
            arr[size - 1] = 0;
            right --;
            left -= 2;
        }

        while(right >= 0) // 从右往左复写
        {
            if(arr[right] != 0)
            {
                arr[left--] = arr[right--];
            }
            else
            {
                arr[left--] = 0;
                arr[left--] = 0;
                right--;
            }
        }
    }
};