集合栈

题目描述

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

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

思路：
1、进栈，定义一个临时数组保存每次要进栈的元素，每当该数组满后，将数组压入集合栈，其中数组大小为每个栈的大小，数组入栈后，清空数组，从头开始。
2、出栈，若临时数组不为空，直接从临时数组出栈，若为空，从集合栈弹出元素到临时数组，从临时数组出栈。

class SetOfStacks {
public:
    vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size) {
        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 (temp.size() != 0)
            st.push_back(temp);
        return st;
    }
};