集合栈

对入栈和出栈操作,top都需要相应加减,对一些条件的判断。

/*****************************************************
* file SetOfStacks.cpp
* date 2016/05/05 17:49

* 问题描述:
请实现一种数据结构SetOfStacks,由多个栈组成,其中每个栈的大小为size,当前一个栈填满时,新建一个栈。
该数据结构应支持与普通栈相同的push和pop操作。
给定一个操作序列int[][2] ope(C++为vector<vector<int>>),每个操作的第一个数代表操作类型,
若为1,则为push操作,后一个数为应push的数字;若为2,则为pop操作,后一个数无意义。
请返回一个int[][](C++为vector<vector<int>>),为完成所有操作后的SetOfStacks,顺序应为从下到上,默认初始的SetOfStacks为空。保证数据合法。

* 问题分析:
这道题目基本没什么技巧,就是堆栈入栈的过程,相当于用多个数组来实现一个堆栈,如果一个数组满了,
则使用另一数组,如果当且数组空了,则把该数组删除,从它前一个数组中操作,实现的时候注意数组交换数组时应该如何判断。

*****************************************************/

#include <iostream>
using namespace std;
#include <vector>
class SetOfStacks {
public:
    vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size) {
        // write code here
        if (ope.size()==0)
        {
            return ope;
        }
        vector<vector<int>> vec_result;
        vector<int> vec;
        int top = 0;
        for (int i = 0; i < ope.size();i++)
        {
            if (ope[i][0]==1)   //按顺序执行,则不需要if--else
            {
                vec.push_back(ope[i][1]);
                if (++top==size)  //栈满,则新建一个栈
                {
                    vec_result.push_back(vec);
                    vec.clear();
                    top = 0;
                }
            }
            else
            {
                if (top==0)  //栈为空,则从上一个栈出
                {
                    vec = vec_result.back();
                    vec_result.pop_back();
                    top = size;
                }
                vec.pop_back();
                top--;
            }
        }

        //vector<vector<int> > st;
        //vector<int> temp;
        //for (int i = 0; i < ope.size(); i++){
        //    if (ope[i][0] == 1){//入栈
        //        if (temp.size() == size){//栈满,则新建一个栈,再入栈
        //            st.push_back(temp);
        //            temp.clear();
        //            temp.push_back(ope[i][1]);
        //        }
        //        else
        //            temp.push_back(ope[i][1]);
        //    }
        //    if (ope[i][0] == 2){//出栈
        //        if (temp.size() != 0)//当前栈不为空,直接出栈
        //            temp.pop_back();
        //        else if (st.size() != 0){//当前栈为空,则从上一个栈出
        //            temp = st[st.size() - 1];
        //            temp.pop_back();
        //            st.pop_back();
        //        }
        //    }
        //}

        if (!vec.empty())   //对于没有满size但又有元素的vec,最后压入返回结果
        {
            vec_result.push_back(vec);
        }
        return vec_result;
    }
};
原文地址:https://www.cnblogs.com/ranjiewen/p/5462932.html