集合栈

题目描述

请实现一种数据结构SetOfStacks，由多个栈组成，其中每个栈的大小为size，当前一个栈填满时，新建一个栈。该数据结构应支持与普通栈相同的push和pop操作。

给定一个操作序列int[][2] ope(C++为vector<vector<int>>)，每个操作的第一个数代表操作类型，若为1，则为push操作，后一个数为应push的数字；若为2，则为pop操作，后一个数无意义。请返回一个int[][](C++为vector<vector<int>>)，为完成所有操作后的SetOfStacks，顺序应为从下到上，默认初始的SetOfStacks为空。保证数据合法。

----------------------------------------

-----------------------------------------

class SetOfStacks {
public:
    vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size) {
        // write code here
        vector<vector<int> > res;
        vector<int> temp;
        for (decltype(ope.size()) i=0; i<ope.size(); ++i){
            if (ope[i][0] == 1){
                //push
                if (temp.size()==size){
                    //当前栈满，“新建”栈，再push
                    res.push_back(temp);
                    temp.clear();
                    temp.push_back(ope[i][1]);
                }else
                    temp.push_back(ope[i][1]);
            }else {
                //pop
                if (temp.size()){
                    //当前栈不空，直接出栈
                    temp.pop_back();
                }else {
                    //从上一个栈 出栈（连续出栈可能出现）
                    temp = res.back();
                    res.pop_back();
                    temp.pop_back();
                }
            }
        }
        //注意所有序列操作结束后，处理temp
        if (temp.size())
            res.push_back(temp);
        return res;
    }
};