Implement Stack using Queues

LeetCode Q 225 - Implement Stack using Queues

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.

Example:
MyStack stack = new MyStack(); stack.push(1); stack.push(2);
stack.top(); // returns 2
stack.pop(); // returns 2
stack.empty(); // returns false

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).

Similar Question: Implement Queue using Stacks

Solution

Code:


Queue<Integer> que1;
Queue<Integer> que2;

/ ** Initialize your data structure here. * /
public MyStack() {
	que1 = new LinkedList<>();
	que2 = new LinkedList<>();
}

/ ** Push element x onto stack. * /
public void push(int x) {
	que1.offer(x);
}

/ ** Removes the element on top of the stack and returns that element. * /
public int pop() {
	
	while(que1.size() > 1)  
		que2.offer(que1.poll()); 
	
	int ans = que1.poll();
	que1 = que2;
	que2 = new LinkedList<>();
	
	return ans;
}

/ ** Get the top element. * /
public int top() {
	
	while(que1.size() > 1) 
		que2.offer(que1.poll());
	
	int ans = que1.peek();
	que2.offer(que1.poll());
	que1 = que2;
	que2 = new LinkedList<>();
	
	return ans;
}

/ ** Returns whether the stack is empty. * /
public boolean empty() {
	return que1.isEmpty();
}

   Reprint policy


《Implement Stack using Queues》 by Tong Shi is licensed under a Creative Commons Attribution 4.0 International License
  TOC