数组:1089.复写零

题目:

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

思路:借助栈,将arr数组中元素进栈,遇见0要连续多压入1个,当栈元素等于arr的长度时停止压入元素,然后将栈中元素出栈,进入arr中,最后将arr中元素反转(因为栈中元素先进后出)

代码:第一版自己用vector<int> 来模拟队列的功能(先进先出)

class Solution {
public:
    void duplicateZeros(vector<int>& arr) {
        int len = arr.size();
        //用tem模拟队列
        vector<int> tem;
        //初始化tem
        tem.push_back(arr[0]);
        for (int i = 0; i < len; i++) {
            //tem中第一个元素是要插入i位置上的数
            if (tem[0] != 0) {
                 //不为0直接将tem首元素插入i位置
                arr[i] = tem[0];
                //当到arr数组最后一个元素时,后面的数就都不要了
                if ((i + 1) < len) {
                     //将下一个元素压入
                    tem.push_back(arr[i + 1]);
                }
                //删除首元素
                tem.erase(tem.begin());
                continue;
            }
            //当等于0时要同时将两个元素压入,因为arr在i,i+1位置都要插入0
            if ((i + 1) < len) {
                tem.push_back(arr[i + 1]);
                
            }
            if ((i + 2) < len) {
                tem.push_back(arr[i + 2]);
            }
            arr[i] = 0;
            if ((i + 1) < len) {
                arr[i + 1] = 0;
            }
            
            tem.erase(tem.begin());
            i++;

            
        }
    }
    

};

第二版,用栈实现

class Solution {
public:
   void duplicateZeros(vector<int>& arr) {
    stack<int> st;
    //控制栈中元素结束插入,因为栈中元素是最后插入到arr中的元素
    int temp = 0;
    for (int i = 0; i < arr.size(); ++i) {
        st.push(arr[i]);
        ++temp;
        if (arr[i] == 0) {
            if (temp == arr.size())
                break;
            ++temp;
            st.push(0);
        }
        if (temp == arr.size())
            break;
    }

    arr.clear();
    while (!st.empty()) {
        arr.push_back(st.top());
        st.pop();
    }
    //栈是先进后出要反转最后
    reverse(arr.begin(), arr.end());
}

    

};

第一版是内存动态的分配,第二版是栈的大小为arr.size();前者内存好点,但时间慢

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值