栈和队列的相互转化

与世无争的帅哥 提交于 2020-02-12 02:18:36

用桟实现队列
用两个栈实现队列
算法思路

1.入队列:将元素放置到s1中
2.出队列:检测s2是否为空,若为空,将s1中的元素搬移到s2中,删除s2栈顶的元素;若不为空,则删除s2栈顶的元素
3.获取队头元素:检测s2是否为空,若为空,将s1中元素搬移到s2中;若不为空,则从s2栈顶直接获取
4.检测队列是否为空:若两个栈都为空,则队列为空

Java代码

class MyQueue {
    private Stack<Integer> s1;//模拟入队列
    private Stack<Integer> s2;//模拟出队列

    /** Initialize your data structure here. */
    public MyQueue() {
        s1 = new Stack<>();
        s2 = new Stack<>();
    }
    
    /** Push element x to the back of queue. */
    public void push(int x) {
        s1.push(x);
    }
    
    /** Removes the element from in front of queue and returns that element. */
    public int pop() {
        if(s2.isEmpty()){
            while(!s1.isEmpty()){
                s2.push(s1.pop());
            }  
        }
        return s2.pop();
    }
    
    /** Get the front element. */
    public int peek() {
        if(s2.isEmpty()){
            while(!s1.isEmpty()){
                s2.push(s1.pop());
            }
        }
        return s2.peek();
    }
    
    /** Returns whether the queue is empty. */
    public boolean empty() {
        return s1.isEmpty() && s2.isEmpty();
    }
}

用队列实现栈

算法思路

1.入栈:将元素入队列到q1中
2.出栈:
①将q1中元素移动到q2中
②将q1中剩余的一个元素删除掉
③交换q1和q2
3.获取栈顶元素:
①将q1中元素移动到q2中
②从q1中取栈顶元素
③将q1中的一个元素搬移到q2中
④交换q1和q2
4.判空:q2为空

Java代码

class MyStack {
    private Queue<Integer> q1;
    private Queue<Integer> q2;

    /** Initialize your data structure here. */
    public MyStack() {
        q1 = new LinkedList<>();
        q2 = new LinkedList<>();
    }
    
    /** Push element x onto stack. */
    public void push(int x) {
        q1.offer(x);
    }
    
    /** Removes the element on top of the stack and returns that element. */
    public int pop() {
        //将q1中size-1个元素搬移到q2中
        while(q1.size() > 1){
            q2.offer(q1.poll());
        }
        //删除q1中的队头元素
        int ret = q1.poll();
        Queue<Integer> temp = q1;
        q1 = q2;
        q2 = temp;
        return ret;
    }
    
    /** Get the top element. */
    public int top() {
        //将q1中size-1个元素搬移到q2中
        while(q1.size() > 1){
            q2.offer(q1.poll());
        }
        int ret = q1.peek();
        q2.offer(q1.poll());
        Queue<Integer> temp = q1;
        q1 = q2;
        q2 = temp;
        return ret;
    }
    
    /** Returns whether the stack is empty. */
    public boolean empty() {
        return q1.isEmpty();
    }
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!