3.3集合栈

题目描述

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

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

 

学习总结:1.最后存储到的栈是好几个组成的栈,所以入栈考虑是否满了,出栈要考虑是否为空,其他均正常操作

              2.二维vector的操作,size()函数返回的是一维行的大小

class SetOfStacks {
public:
    vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size) {
        // write code here
        //第二个为1,入栈操作,push
        //为2,出栈操作,pop
        //如何判断ope的第二个数字的值
        //没有理解好几个栈的意思,返回的结果是好几个栈组合成的结果
        int s = ope.size();
        vector<vector<int> > result;
        vector<int> temp;
        for(int i = 0; i < s; i++){
            if(ope[i][0] == 1){
                if(temp.size() == size){
                   result.push_back(temp);
                   temp.clear();
                   temp.push_back(ope[i][1]);
                }
                else{
                    temp.push_back(ope[i][1]);
                }   
            }  //入栈操作
            else{
                if(temp.size() == 0){ //这个该怎么操作?
                    temp = result[result.size()-1];//这步是关键
                    temp.pop_back();
                    result.pop_back();
                    
                }
                else temp.pop_back();
             }//出栈操作
            
        }
        if(temp.size() != 0) result.push_back(temp);
        return result;
        
    }
};

 

posted @ 2016-04-11 16:55  梦幻之海  阅读(194)  评论(0编辑  收藏  举报