您的位置:首页 > 产品设计 > UI/UE

225. Implement Stack using Queues

2016-02-29 21:05 501 查看
题目:

Implement the following operations of a stack using queues.
push(x) -- Push element x onto stack.
pop() -- Removes the element on top of the stack.
top() -- Get the top element.
empty() -- Return whether the stack is empty.
Notes:
You must use only standard operations of a queue -- which means only 
push
to back
peek/pop from front
size
,
and 
is empty
 operations are valid.
Depending on your language, queue may not be supported natively. You may simulate a queue by using a list or deque (double-ended queue), as long as you use only standard operations of a queue.
You may assume that all operations are valid (for example, no pop or top operations will be called on an empty stack).
思路:

用两个队列去模拟一个栈,和之前的用两个栈去模拟一个队列很相似,关键在于pop操作的模拟。

我的思路是保证q1是主要的队列,q2是辅助队列。

每次push时都是插入到q1中,当需要pop时,将q1中除了最后一个元素的所有元素remove并push到q2,然后remove最后一个元素,最后再将q1复制成q2。

代码:

class MyStack {
public List<Integer> q1 = new ArrayList<Integer>();
public List<Integer> q2 = new ArrayList<Integer>();

// Push element x onto stack.
public void push(int x) {
q1.add(x);
}

// Removes the element on top of the stack.
public void pop() {
if(q1.size() == 0){return;}
if(q1.size() == 1){
q1.remove(0);
}else{
while(q1.size() > 1){
q2.add(q1.get(0));
q1.remove(0);
}
q1.remove(0);

q1.addAll(q2);
q2.clear();
}
}

// Get the top element.
public int top() {
if(q1.size() == 1){
return q1.get(0);
}else{
while(q1.size() > 1){
q2.add(q1.get(0));
q1.remove(0);
}
int value = q1.get(0);
q2.add(q1.get(0));
q1.remove(0);

q1.addAll(q2);
q2.clear();
return value;
}
}

// Return whether the stack is empty.
public boolean empty() {
return (q1.size()==0 && q2.size()==0);
}
}

之后看了discuss,看到很多种解法,主要使用List的poll和peek方法,由于我对这两个函数用的不多,因此写出来的代码更难看。。。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  leetcode java