LeetCode题5：用两个栈实现队列（Java实现）

双栈实现队列

最新推荐文章于 2024-08-19 09:12:14 发布

原创最新推荐文章于 2024-08-19 09:12:14 发布 · 242 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#栈 #java #队列 #数据结构

leetcode 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法，通过定义两个泛型栈，一个用于添加元素，另一个用于移除元素。当从队头删除节点时，如果第二个栈为空，则将第一个栈的所有元素依次弹出并压入第二个栈，最后弹出第二个栈的栈顶元素即为队头元素。

用两个栈实现队列，实现两个函数appendtail和deleteHead,分别完成在队尾插入节点和队头删除节点的功能

Java语言实现

import java.util.Stack;

/**
 * 用两个栈实现队列
 * 用两个栈实现队列，分别完成队头删节点，队尾加节点
 */
public class problem5 <T>{
  private Stack<T> stack1 =  new Stack<T>();
  private Stack<T> stack2 =  new Stack<T>();
  public void appendTail( T t){
      stack1.push(t);
      for (T t1 : stack1) {
          System.out.println("stack1:"+t1);
      }
  }
  public T deleteHead () {
      if(stack2.isEmpty()){
          if (!stack1.isEmpty()){
              stack2.push(stack1.pop());
              for (T s :stack2) {
                  System.out.println("stack2:"+s);
              }
          }else {
              try {
                  throw new Exception("队列为空，不能删除") ;
              } catch (Exception e) {
                  e.printStackTrace();
              }
          }
      }
     return stack2.pop();

  }

    public static void main(String[] args) {
        problem5<String> P5 = new problem5<>();
        P7.appendTail("1");
        P7.appendTail("2");
        P7.appendTail("3");
        P7.deleteHead();
    }
}