设想有一堆盘子,堆太高可能会倒下来。因此,现实生活中,盘子堆到一定高度时,我们就会另外堆一堆盘子。请事先设计数据结构SetOfStacks, 模拟这种行为。
SetOfStacks应该由多个栈组成,并且在前一个栈填满时新建一个栈。此外,SetOfStacks.push() 和SetOfStacks.pop()应该与普通栈的操作方法相同(也就是说,
pop()返回的值,应该跟只有一个栈时的情况一样)。
import java.util.Stack;
public class SetOfStacks {
class stack {
public void push(int v);
public int pop();
}
//push and pop are both operated from the last stack, just pay attention if the last stack is empty or full
public void push(int value) {
Stack<E> last = getLastStack();
if(last != null && !last.isFull()) {
last.push(value);
}
else {
Stack stack = new Stack(capacity);
stack.push(value);
stacks.add(stack);
}
}
public int pop() {
Stack last = getLastStack();
int v = last.pop();
if (last.size == 0 ) {
stacks.remove(stacks.size() -1);
}
return -1;
}
}
进阶:实现一个popAt(int index)方法,根据指定的子栈,进行pop操作。从栈1弹出元素时,我们需要移除栈2的栈底元素,并将其推到栈1中;随后,将栈3的栈底元素推入栈2,将栈4的元素推入栈3,以此类推