题目描述
请实现一种数据结构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; } };