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

题目描述:

请实现一种数据结构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 here
int 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;
}
};