剑指offer面试题9–两个栈实现队列
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
刚开始两个栈都是空的,要进队列,可以将元素压入任何一个栈,不妨就压入stack1中,我们知道元素在栈中是后进先出的,因此要出列需要删除stack1最底部的元素,此时stack2派上用场了,将stack1的元素再压入stack2中,现在元素的排列变成了原来插入的顺序,要出列的元素到了栈顶,要完成出列操作就很方便了。若要继续出列,只要stack2不为空,只需继续从stack2中出栈即可,**因为stack2中的栈顶元素就是最先入列的元素。若stack2为空,就将stack1中的元素压入到stack2后,再弹出栈顶元素。**如果要入列,就直接压入stack1中。
总结一下就是:
- stack1专门用于进入队列
- stack2专门用于出队列
- 出列操作。stack2不为空就直接出列;为空就将stack1中所有元素压入stack2中,再弹出栈顶元素。
package com.offer._9;
import java.util.LinkedList;
/**
* @author :jhys
* @date :Created in 2020/11/26 16:04
* @Description :
*/
public class Solution {
private LinkedList<Integer> stack1 = new LinkedList<>();
private LinkedList<Integer> stack2 = new LinkedList<>();
public void enqueue(int node) {
// stack1专门用于进入队列
stack1.push(node);
}
// 出列操作。stack2不为空就直接出列;为空就将stack1中所有元素压入stack2中,再弹出栈顶元素。
public int dequeue() {
if (stack1.isEmpty() && stack2.isEmpty()) {
throw new RuntimeException("队列为空");
}
if (stack2.isEmpty()) {
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}