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

用两个栈实现队列

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

基本思路

在进队操作时用stack1接收进队元素,在出队操作时将栈中的元素全部推到stack2,实现元素的反序,这样stack2中栈顶弹出的元素就是最先进来的元素,但是有可能出现连续进队后中间有出队再进队的情况,例如stack1进队[2,3],出队时stack2中元素为[3,2],将2出队,此时如果没有将stack2中的元素再推回stack1,后面再进一个1,再进行出队操作时,stack2中的元素为[3,1],出队的元素就是1而不是2了,因此还需要把另一个栈的元素再推回原来的栈中

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() {
        int ret = 0;
        while(!stack1.isEmpty()) {
            stack2.push(stack1.pop());
        }
        if(!stack2.isEmpty()) {
            ret = stack2.pop();
        }
        // 将第二个栈的元素再放回第二个栈,这样才能处理连续进队中间有出队的情况
        while(!stack2.isEmpty()) {
            stack1.push(stack2.pop());
        }
        return ret;
    }
}

全部评论

相关推荐

点赞 评论 收藏
分享
07-24 16:39
已编辑
门头沟学院 测试开发
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务