标签:
题目
正如标题所述,你需要使用两个栈来实现队列的一些操作。
队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。
pop和top方法都应该返回第一个元素的值。
比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2
仅使用两个栈来实现它,不使用任何其他数据结构,push,pop 和 top的复杂度都应该是均摊O(1)的
解题
两个栈stack1 、stack2,一个存储入队列元素,一个存储出出队列元素
stack2 入队列
stack1出队列
当入队列的时候:直接在入栈stack2
当出队列的时候:若栈stack2中有元素,则栈stack2底的元素就是所要出队列的元素,将栈stack2中的元素全部出来,并放到栈stack1中,此处stack1栈顶元素就是答案
这里有个问题是若栈stack1中有元素,则就直接取出栈stack1的栈顶元素就是答案
Java
public class Queue { private Stack<Integer> stack1; private Stack<Integer> stack2; public Queue() { // do initialization if necessary stack1 = new Stack<Integer>(); stack2 = new Stack<Integer>(); } private void stack2Tostack1(){ while(!stack2.empty()){ stack1.push(stack2.peek()); stack2.pop(); } } public void push(int element) { // write your code here stack2.push(element); } public int pop() { // write your code here if( stack1.empty()){ this.stack2Tostack1(); } return stack1.pop(); } public int top() { // write your code here if(stack1.empty()){ this.stack2Tostack1(); } return stack1.peek(); } }
Python
class MyQueue: def __init__(self): self.stack1 = [] self.stack2 = [] def push(self, element): # write your code here self.stack2.append(element) def top(self): # write your code here # return the top element self.stack2Tostack1() return self.stack1[len(self.stack1)-1] def pop(self): # write your code here # pop and return the top element self.stack2Tostack1() return self.stack1.pop() def stack2Tostack1(self): if len(self.stack1)==0: while len(self.stack2)!=0: self.stack1.append(self.stack2.pop())
参考了九章 ,由于初始化不知道怎么定义
lintcode 中等题:implement queue by two stacks 用栈实现队列
标签:
原文地址:http://www.cnblogs.com/theskulls/p/5103516.html