欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

剑指offer——用两个栈实现队列

程序员文章站 2022-07-10 10:26:39
...

1,问题描述

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

2,解题思路

(1)明确栈和队列的含义

  • 栈是先进后出
  • 队列是先进先出

(2)用stack1做入队操作,这一步没有需要特别注意的地方,来一个入一个就行

(3)用stack2做出队操作,基本思想是先把stack1中的元素依次弹出放到stack2中,再把stack2中的元素弹出即可。但是需要注意以下几点:

  • 在stack2执行弹出操作时,必须判断该栈中是否还有元素,如果有,则先弹出该栈中的元素,否则,进行下一步操作
  • 把stack1中的元素依次弹出,并依次放入stack2中,再实行stack2的弹出操作

3,源码

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        stack1.push(node);
    }
    
    public int pop() {
        if(!stack2.empty()){
            return stack2.pop();
        }else{
            while(!stack1.empty()){
            stack2.push(stack1.pop());
            }
            return stack2.pop();
        }

    }
}