剑指 Offer 09. 用两个栈实现队列
用
两个栈
实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead
,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )示例 1: 输入:
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出:[null,null,3,-1]示例 2: 输入:
[“CQueue”,“deleteHead”,“appendTail”,“appendTail”,“deleteHead”,“deleteHead”][[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]链接:https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
解析题目:
输入:
第一行为执行的方法
第二行为对应方法的参数,其下标与第一行一一对应
输出:
对应下标方法的返回值,CQueue与appendTail返回值均为null。
deleteHead返回值为队列头部被删除的元素(若队列中没有元素,deleteHead 操作返回 -1 )。
思路:
使用Java类 LinkedList<>,链接列表可用作堆栈、队列或双端队列。
- void push(E e) :将元素推入此列表所表示的堆栈。
- E pop() :从此列表所表示的堆栈处弹出一个元素。
创建两个栈s_in和s_out。
appendTail()执行,元素进入in队列。
栈有先进后出的特性。
将in中元素全部弹出并进入out。此使in弹出的第一个元素是最后一个进入in中的。此元素将是out第一个进入的,之后将是最后一个弹出的。
这样保证了删除的是队列第一个元素。
代码
class CQueue {
LinkedList s_in;
LinkedList s_out;
public CQueue() {
s_in = new LinkedList();
s_out = new LinkedList();
}
public void appendTail(int value) {
s_in.push(value);
}
public int deleteHead() {
if (s_out.isEmpty()){
while (!s_in.isEmpty()){
s_out.push(s_in.pop());
}
}
if (s_out.isEmpty()){
return -1;
}else{
return (int) s_out.pop();
}
}
}