《程序员面试金典》--求集合栈

来源:互联网 发布:软件 电脑 地图 编辑:程序博客网 时间:2024/04/30 03:18
题目描述:
请实现一种数据结构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>#include<vector>using namespace std;class SetOfStacks {public:    vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size) {        // write code hereint sz=ope.size();if(sz==0)return ope;vector<vector<int> > vecvec;vector<int> vec;int top=0;for(int i=0;i<sz;i++){if(ope[i][0]==1){vec.push_back(ope[i][1]);if(++top==size){vecvec.push_back(vec);vec.clear();top=0;}}else{if(top==0){vec=vecvec.back();vecvec.pop_back();top=size;}vec.pop_back();top--;}}if(!vec.empty())vecvec.push_back(vec);return vecvec;    }};


0 0