【剑指offer】面试题07:用两个栈实现队列

xiaoxiao2021-02-28  123

题目

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

思路

栈:先进后出队列:先进先出   其实知道了栈和队列的特点这道题就很容易了,思路也就不多说了,直接看代码实现就可以理解。

代码实现

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(stack1.isEmpty()&&stack2.isEmpty()){ throw new RuntimeException("Queue is empty!"); } //stack2为空时,将stack1中的元素压入stack2 //stack2不为空时,新压入stack1的元素是无法被压入stack2的,保证了出队列的顺序 if(stack2.isEmpty()){ while(!stack1.isEmpty()){ stack2.push(stack1.pop()); } } return stack2.pop(); } }

注:运行环境——牛客网OJ。

转载请注明原文地址: https://www.6miu.com/read-17365.html

最新回复(0)