请实现一种数据结构SetOfStacks,由多个大小为size的栈组成,当前一个栈填满时,则新建一个栈,且也可以与普通栈一样拥有相同的push和pop操作。
现给定一个操作序列int[][2] ope(C++为vector<vector<int>>),若执行push操作则第一个数为1,第二个数为应push的数字;若执行pop操作,则第一个数为2,第二个数为空。返回值为int[][](C++为vector<vector<int>>),即为变动后的SetOfStacks,顺序从下到上,初始SetOfStacks为空,并保证数据合法。
public:
vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size) {
// write code here
vector<int> vect;
vector<vector<int>> A;
int len = vect.size();
for(int i = 0; i < ope.size(); i ++)
{
//压入操作时
if(ope[i][0] == 1)
{
//判断A是否为空
if(A.size() == 0)
{
//判断vect是否已满
if(vect.size() == size)
{
A.push_back(vect);
vect.clear();
vect.push_back(ope[i][1]);
}
else
vect.push_back(ope[i][1]);
}
else //A不为空
{
//判断A的顶栈是否已满
if(A[A.size() - 1].size() < size)
A[A.size() - 1].push_back(ope[i][1]);
else
{
if(vect.size() == size)
{
A.push_back(vect);
vect.clear();
vect.push_back(ope[i][1]);
}
else
vect.push_back(ope[i][1]);
}
}
}
else //弹出操作
{
//判断vect是否为空
if(vect.empty()) //为空
{
//弹出A顶栈的元素
A[A.size() - 1].pop_back();
//判断顶栈是否为空
if(A[A.size() - 1].empty()) //为空则弹出顶栈
A.pop_back();
}
else //vect不为空
vect.pop_back();
}
}
if(!vect.empty())
A.push_back(vect);
return A;
}
};