版权声明:本文为博主原创文章,转载请标明出处。 https://blog.csdn.net/weixin_43014205/article/details/85951991
题目:用两个栈实现一个队列,请实现它的两个函数appendTail和DeleteHead,分别完成在队列尾部插入节点和在队列头部删除节点的功能。
思路:两个栈,记为stack1和stack2,把元素入栈到stack1,再出栈压入到stack2,这样stacks中的元素由栈顶到栈底的顺序就和队列相同了,然后stack2依次出栈,相当于DeleteHead,那该如何实现appendTail,就是想stack1中push元素
实现DeleteHead的过程中,一定要当stack2为空时,再把stack1中的元素出栈压入到stack2,再就是当stack2为空时,要把当前stack1中的所有元素pop,在push到Stack2中,这样就完成了appendTail和DeleteHead的统一。
只要stack1和stack2至少有一个不为空,就可以DeleteHead
//定义Node
public class Node {
int value;
public Node(int value) {
this.value = value;
}
@Override
public String toString() {
return "Node{" +
"value=" + value +
'}';
}
}
import java.util.Stack;
public class PushAndPop {
static Stack<Node> stack1 = new Stack<>();
static Stack<Node> stack2 = new Stack<>();
//往stack1入栈 相当于在队尾插入
public static void appendTail(Node node){
stack1.push(node);
}
//从栈2出栈 相当于出队
public static Node deleteHead() throws Exception {
//如果两个栈都为空 抛出异常
if(stack1.isEmpty() && stack2.isEmpty()){
throw new Exception("队列为空");
}
//把stack1中的所有元素都pop且push到Stack2中
if(stack2.isEmpty()) {
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
public static void main(String[] args) throws Exception {
PushAndPop.appendTail(new Node(1));
PushAndPop.appendTail(new Node(2));
System.out.println(PushAndPop.deleteHead());
System.out.println(PushAndPop.deleteHead());
PushAndPop.appendTail(new Node(3));
System.out.println(PushAndPop.deleteHead());
}
}
/**
*Node{value=1}
*Node{value=2}
*Node{value=3}
*/