【 声明:版权所有,转载请标明出处,请勿用于商业用途。 联系信箱:libin493073668@sina.com】
题目描述
请实现一种数据结构SetOfStacks,由多个栈组成,其中每个栈的大小为size,当前一个栈填满时,新建一个栈。该数据结构应支持与普通栈相同的push和pop操作。
给定一个操作序列int[][2] ope(C++为vector<vector<int>>),每个操作的第一个数代表操作类型,若为1,则为push操作,后一个数为应push的数字;若为2,则为pop操作,后一个数无意义。请返回一个int[][](C++为vector<vector<int>>),为完成所有操作后的SetOfStacks,顺序应为从下到上,默认初始的SetOfStacks为空。保证数据合法。
思路
因为这里是使用vector来实现这个栈的,所以相对就容易了,首先我们要注意的是,当辅助栈被清空的时候,我们要注意对于答案vector的处理
class SetOfStacks
{
public:
vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size)
{
// write code here
vector<vector<int> > ans;
int len = ope.size();
if(len==0 || size<=0)
return ans;
vector<int> tmp;
for(int i = 0; i<len; i++)
{
if(ope[i][0]==1)
{
if(tmp.size()==size)
{
ans.push_back(tmp);
tmp.clear();
}
tmp.push_back(ope[i][1]);
}
else
{
if(tmp.size()==0)
{
int ansSize = ans.size();
tmp = ans[ansSize-1];
ans.pop_back();
}
tmp.pop_back();
}
}
if(tmp.size())
ans.push_back(tmp);
return ans;
}
};