CC14:集合栈
题目
请实现一种数据结构SetOfStacks,由多个栈组成,其中每个栈的大小为size,当前一个栈填满时,新建一个栈。该数据结构应支持与普通栈相同的push和pop操作。
给定一个操作序列int[][2] ope(C++为vector<vector
请返回一个int[],为完成所有操作后的SetOfStacks,顺序应为从下到上,默认初始的SetOfStacks为空。保证数据合法。
解法
这道题表述的不是特别清楚,举个例子:
1 2 1 3 1 4
2 2 1 5 2 4-------->表示的意思依次是:插入2、插入3、插入4、消除4(栈顶元素)、插入5、消除5(栈顶元素)、插入6、插入7、消除7
1 6 1 7 2 0
所以我们只需要依次判断数组元素,如果是1的话后面跟着的就是要插入的数字,如果是2的话就将之前插入的数字排除。而这里还涉及到了栈的size问题,由于返回类型为vector<vector<int>>
,我们实际上可以利用一个vector类型(vec)来模拟栈。当vec的大小为size的时候就将vec存入到vector<vector
整个程序的逻辑还是非常清楚的,流程化思维就可以解决这个问题,AC代码如下:
class SetOfStacks {
public:
vector<vector<int> > setOfStacks(vector<vector<int> > ope, int size) {
// write code here
vector<vector<int> > res;
if(ope.size()==0)
return res;
vector<int> vec;
for(int i=0;i<ope.size();i++)
{
if(ope[i][0]==1)
{
vec.push_back(ope[i][1]);
if(vec.size()==size)
{
res.push_back(vec);
vec.clear();
}
}
else{
if(vec.size()==0)
{
vec=res.back();
res.pop_back();
}
vec.pop_back();
}
}
if(!vec.empty())
res.push_back(vec);
return res;
}
};
作者:YunLambert
-------------------------------------------
个性签名:一名会音乐、爱健身的不合格程序员
可以Follow博主的Github哦(っ•̀ω•́)っ✎⁾⁾