package exercise1;
import java.util.Stack;
public class Solution {
Stack stack1 = new Stack();
Stack stack2 = new Stack();
public void push(int node) {
stack1.push(node);
}
public int pop() {
while(!stack1.empty()){
stack2.push(stack1.pop());
}
int first = stack2.pop();
while(!stack2.empty()){
stack1.push(stack2.pop());
}
return first;
}
public static void main(String[] args) {
Solution solution=new Solution();
solution.push(1);
solution.push(2);
solution.push(3);
solution.push(4);
System.out.println(solution.pop());
solution.push(2);
System.out.println(solution.pop());
System.out.println(solution.pop());
System.out.println(solution.pop());
System.out.println(solution.pop());
}
}
很容易想到把数据压入到第一个栈然后弹出到第二个栈,这时弹出第二个栈就能实现先进先出了。
这里pop里肯定要这么写:
while(!stack1.empty()){
stack2.push(stack1.pop());
}
但是之后不能就不管了直接return stack.pop();
因为如果弹出一个后再压入一个元素,是压入stack1中,而此时1已经空了,会把元素压入到2的顶端,从而出问题。
所以stack1肯定是要维护着的,而stack2只要1的尾端的值弹出就好了,剩下的值重新给回stack1,以便新的值压入。