题解 | #用两个栈实现队列#

用两个栈实现队列

https://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6

import java.util.*;
import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        stack1.push(node);
    }
    
    public int pop() {
        if(!stack2.isEmpty()){
            return stack2.pop();
        }else{
            if(stack1.isEmpty()){
                return -1;
            }
            while(!stack1.isEmpty()){
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}

核心逻辑就是搞清楚什么时候才可以pop;队列是先进先出,栈是先进后出,

那么我们在pop的时候就要分两种状况:

1.当stack2不为空时,栈中的这些数据是之前留下的,也就是先加入的,根据先进先出,所以我们就可以直接将其pop()出来,

2.当stack2为空,stack1不为空时,那么stack1中的数据就都是后加入的数据,根据栈的先进后出,我们就需要将stack1中的数据全部遍历到stack2中取出第一个数据,该数据就是最先加入的数据了,当然如果stack1中没有数据,那么说明队列为空了,这时候pop()就会报错。

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务