LeetCode 225:用队列实现栈

用队列实现栈

使用队列实现栈的下列操作：

push(x) – 元素 x 入栈
pop() – 移除栈顶元素
top() – 获取栈顶元素
empty() – 返回栈是否为空

注意:

你只能使用队列的基本操作-- 也就是 push to back, peek/pop from front, size, 和 is empty 这些操作是合法的。你所使用的语言也许不支持队列。 你可以使用 list 或者 deque（双端队列）来模拟一个队列 , 只要是标准的队列操作即可。你可以假设所有操作都是有效的（例如, 对一个空的栈不会调用 pop 或者 top 操作）。

public class TestDemo2 {
    public Queue<Integer> queue1=new LinkedList<>();
    public Queue<Integer> queue2=new LinkedList<>();

    //元素入栈
    public void push(int x){
        if(!queue1.isEmpty()){
            queue1.offer(x);
        }else if(!queue2.isEmpty()){
            queue2.offer(x);
        }else {
            queue1.offer(x);
        }
    }

    //移除栈顶元素，并返回栈顶元素的值
    public int pop(){
        if(!queue1.isEmpty()){
            int size1=queue1.size()-1;
            for (int i = 0; i <size1 ; i++) {
                queue2.offer(queue1.poll());
            }
            return queue1.poll();
        }

        if(!queue2.isEmpty()){
            int size2=queue2.size()-1;
            for (int i = 0; i < size2; i++) {
                queue1.offer(queue2.poll());
            }
            return queue2.poll();
        }
        return -1;
    }

    //得到栈顶元素
    public int top(){
        if(!queue1.isEmpty()){
            int temp1=0;
            int size1=queue1.size();
            for (int i = 0; i < size1; i++) {
                temp1=queue1.poll();
                queue2.offer(temp1);
            }
            return temp1;
        }

        if(!queue2.isEmpty()){
            int temp2=0;
            int size2=queue2.size();
            for (int i = 0; i < size2; i++) {
                temp2=queue2.poll();
                queue1.offer(temp2);
            }
            return temp2;
        }
        return -1;
    }

    //判断是否为空
    public boolean empty(){
        if(queue1.isEmpty() && queue2.isEmpty()){
            return true;
        }
        return false;
    }

    public static void main(String[] args) {
        TestDemo2 testDemo2=new TestDemo2();
        testDemo2.push(1);
        testDemo2.push(6);
        testDemo2.push(12);
        testDemo2.push(24);
        testDemo2.push(34);
        testDemo2.push(52);
        System.out.println(testDemo2.pop());
        System.out.println(testDemo2.top());
    }
}