leetcode笔记之[1089. 复写零]

82 阅读1分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述:

1089. 复写零 - 力扣(LeetCode) (leetcode-cn.com)

给你一个长度固定的整数数组 arr,请你将该数组中出现的每个零都复写一遍,并将其余的元素向右平移。

注意:请不要在超过该数组长度的位置写入元素。

要求:请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。

 

示例 1:

输入:[1,0,2,3,0,4,5,0]
输出:null
解释:调用函数后,输入的数组将被修改为:[1,0,0,2,3,0,0,4]

示例 2:

输入:[1,2,3]
输出:null
解释:调用函数后,输入的数组将被修改为:[1,2,3]

提示:

  • 1 <= arr.length <= 10000
  • 0 <= arr[i] <= 9

二、思路分析:

先从最后一个元素开始把数组中的元素压入栈中 然后清空数组 再从栈顶元素开始往数组中添加元素,添加过程中,当栈顶元素为0时需要一次添加两个0 整个过程需要另一个变量len:arr.size()维护, 在栈顶为零的时候需要判断数组元素是否还能加得进两个元素

三、AC 代码:

class Solution {
public:
    void duplicateZeros(vector<int>& arr) {
        stack<int> stk;
        int len = arr.size();
        for (int i = len - 1; i >= 0; i --)
            stk.push(arr[i]);
        arr.clear();
        while(len--)
        {
            if(stk.top() != 0)
            {
                arr.push_back(stk.top());
                stk.pop();
            }
            else
            {
                stk.pop();
                arr.push_back(0);
                if(len--)
                    arr.push_back(0);
                else 
                    break;
            }
        }
    }
};

范文参考

复写零:栈 - 复写零 - 力扣(LeetCode) (leetcode-cn.com)

复写0 - 复写零 - 力扣(LeetCode) (leetcode-cn.com)