用栈实现队列(java)

2024-01-23 09:12:14 浏览数 (2)

仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(pushpoppeekempty):

实现 MyQueue 类:

  • void push(int x) 将元素 x 推到队列的末尾
  • int pop() 从队列的开头移除并返回元素
  • int peek() 返回队列开头的元素
  • boolean empty() 如果队列为空,返回 true ;否则,返回 false
代码语言:javascript复制
class MyQueue {
//首先创建两个栈
    private Stack<Integer> s1;
    private Stack<Integer> s2;
    public MyQueue() {
        s1 = new Stack<>();
        s2 = new Stack<>();
    }
    
    public void push(int x) {
        s1.push(x);//入队:把数据放入第一个栈中
    }
    
    public int pop() {//出队:出s2这个栈中的栈顶元素,如果s2为空,把s1中的所有元素都放入s2中
        if(empty()) {
            return -1;
        }
        if (s2.empty()) {
            while(!s1.empty()) {
                s2.push(s1.pop());
            }
        }
        return s2.pop();
    }
    
    public int peek() {
         if(empty()) {
            return -1;
        }
        if (s2.empty()) {
            while(!s1.empty()) {
                s2.push(s1.pop());
            }
        }
        return s2.peek();
    }
    
    public boolean empty() {//当两个栈都为空说明队列为空
        return (s1.isEmpty() && s2.isEmpty());
    }
}

0 人点赞