QueueWithTwoStack.java

package datastructure.queue;

import org.junit.Test;

import java.util.Stack;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;

public class QueueWithTwoStack {

 /*
 TASK
 Stack 두 개로 Queue를 구현한다.
 */

 @Test
 public void test() {
 MyQueue<Integer> queue = new MyQueue<>();
 queue.offer(1);
 queue.offer(2);
 queue.offer(3);
 queue.offer(4);

 assertThat(queue.size(), is(4));

 assertThat(queue.poll(), is(1));
 assertThat(queue.poll(), is(2));
 assertThat(queue.poll(), is(3));
 assertThat(queue.poll(), is(4));

 assertThat(queue.size(), is(0));
 }

 public class MyQueue<T> {
 private Stack<T> stack1;
 private Stack<T> stack2;

 public MyQueue() {
 stack1 = new Stack<T>();
 stack2 = new Stack<T>();
 }

 public void offer(T elm) {
// 1. 아예 stack2를 queue와 동일하게 데이터를 저장하는 방법
// while (!stack2.isEmpty()) {
// stack1.push(stack2.pop());
// }
// stack1.push(elm);
//
// while (!stack1.isEmpty()) {
// stack2.push(stack1.pop());
// }

 // 2. poll할 때만 queue처럼 반환하는 방법
 stack1.push(elm);
 }

 public T poll() {
// 1. 아예 stack2를 queue와 동일하게 데이터를 저장하는 방법
// return stack2.pop();

// 2. poll할 때만 queue처럼 반환하는 방법
 if (stack2.isEmpty()) {
 while(!stack1.isEmpty()) {
 stack2.push(stack1.pop());
 }
 }
 return stack2.pop();

 }

 public int size() {
// 1. 아예 stack2를 queue와 동일하게 데이터를 저장하는 방법
// return stack2.size();

// 2. poll할 때만 queue처럼 반환하는 방법
 return stack1.size() + stack2.size();
 }
 }
}