堆盘子。设想有一堆盘子,堆太高可能会倒下来。因此,在现实生活中,盘子堆到一定高度时,我们就会另外堆一堆盘子。请实现数据结构SetOfStacks,模拟这种行为。SetOfStacks应该由多个栈组成,并且在前一个栈填满时新建一个栈。此外,SetOfStacks.push()和SetOfStacks.pop()应该与普通栈的操作方法相同(也就是说,pop()返回的值,应该跟只有一个栈时的情况一样)。 进阶:实现一个popAt(int index)方法,根据指定的子栈,执行pop操作。
当某个栈为空时,应当删除该栈。当栈中没有元素或不存在该栈时,pop,popAt 应返回 -1.
class StackOfPlates {
public:
int n;
vector< stack<int> > v;
StackOfPlates(int cap) {
n = cap;
}
void push(int val) {
if(n>=1){
stack<int> s;
s.push(val);
if(v.empty()){
v.push_back(s);
}
else{
if(v.back().size() < n){
v.back().push(val);
}
else{
v.push_back(s);
}
}
}
}
int pop() {
if(v.empty()){
return -1;
}
else{
int a = v.back().top();
v.back().pop();
if(v.back().empty()){
v.pop_back();
}
return a;
}
}
int popAt(int index) {
if(v.empty() || index >= v.size()){
return -1;
}
else{
int a = v[index].top();
v[index].pop();
if(v[index].empty()){
auto it = v.begin()+index;
v.erase(it);
}
return a;
}
}
};
/**
* Your StackOfPlates object will be instantiated and called as such:
* StackOfPlates* obj = new StackOfPlates(cap);
* obj->push(val);
* int param_2 = obj->pop();
* int param_3 = obj->popAt(index);
*/