40. 用栈实现队列:
正如标题所述,你需要使用两个栈来实现队列的一些操作。
队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。
pop和top方法都应该返回第一个元素的值。
假设调用pop()函数的时候,队列非空
样例 1
输入:
push(1)
pop()
push(2)
push(3)
top()
pop()
输出:
1
2
2
样例 2
输入:
push(1)
push(2)
push(2)
push(3)
push(4)
push(5)
push(6)
push(7)
push(1)
输出:
[]
挑战
仅使用两个栈来实现它,不使用任何其他数据结构,push,pop 和 top的复杂度都应该是均摊O(1)的
原题传送门
题解
public class MyQueue {
private final Stack<Integer> inStack = new Stack<>();
private final Stack<Integer> outStack = new Stack<>();
public MyQueue() {
// do intialization if necessary
}
/*
* @param element: An integer
* @return: nothing
*/
public void push(int element) {
// write your code here
inStack.push(element);
}
/*
* @return: An integer
*/
public int pop() {
// write your code here
adjust();
return outStack.pop();
}
/*
* @return: An integer
*/
public int top() {
// write your code here
adjust();
return outStack.peek();
}
private void adjust() {
if (outStack.isEmpty()) {
while (!inStack.isEmpty()) {
outStack.push(inStack.pop());
}
}
}
}
最后说两句
非常感谢你阅读本文章,如果你觉得本文对你有所帮助,请留下你的足迹,点个赞,留个言,多谢~
作者水平有限,如果文章内容有不准确的地方,请指正。
希望小伙伴们都能每天进步一点点。
本文由二当家的白帽子博客原创,转载请注明来源,谢谢~