Java 剑指offer 面试题9:两个栈实现队列

版权声明:本文为博主原创文章,转载请标明出处。 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}
*/

猜你喜欢

转载自blog.csdn.net/weixin_43014205/article/details/85951991
今日推荐