题目
堆盘子。设想有一堆盘子,堆太高可能会倒下来。因此,在现实生活中,盘子堆到一定高度时,我们就会另外堆一堆盘子。请实现数据结构SetOfStacks,模拟这种行为。SetOfStacks应该由多个栈组成,并且在前一个栈填满时新建一个栈。此外,SetOfStacks.push()和SetOfStacks.pop()应该与普通栈的操作方法相同(也就是说,pop()返回的值,应该跟只有一个栈时的情况一样)。 进阶:实现一个popAt(int index)方法,根据指定的子栈,执行pop操作。
当某个栈为空时,应当删除该栈。当栈中没有元素或不存在该栈时,pop,popAt 应返回 -1.
示例1:
输入: ["StackOfPlates", "push", "push", "popAt", "pop", "pop"] [[1], [1], [2], [1], [], []] 输出: [null, null, null, 2, 1, -1]
示例2:
输入: ["StackOfPlates", "push", "push", "push", "popAt", "popAt", "popAt"] [[2], [1], [2], [3], [0], [0], [0]] 输出: [null, null, null, null, 2, 1, 3]
解题
方法一:模拟
用一个vector来动态地存储stack
capacity来记录容量
在push的时候,如果出现空的或者满的,就创建一个新的栈,再在里面放元素
在pop的时候,如果栈空了,就清理掉。(因此,结合push和pop每个栈都不可能是空的).
class StackOfPlates { public: int capacity; vector<stack<int>> stks; StackOfPlates(int cap) { capacity=cap; } void push(int val) { if(capacity==0) return; if(stks.empty()||stks.back().size()==capacity){ stks.push_back(stack<int>()); } stks.back().push(val); } int pop() { return popAt(stks.size()-1); } int popAt(int index) { if(index>=stks.size()) return -1;//如果capacity为空,那么stks.size()==0,无论index怎么取都不行 int res=stks[index].top(); stks[index].pop(); if(stks[index].empty()){ stks.erase(stks.begin()+index); } return res; } };