双栈双队列实现API接口

栈排序

图片说明

重点在push操作

class SortedStack {

    Stack<Integer> stackMin;
    Stack<Integer> stack;

    public SortedStack() {
        stackMin = new Stack<>();
        stack = new Stack<>();
    }

    public void push(int val) {
        //放入栈中前,将最小的弹出,并放在辅助栈中
        while(!stack.isEmpty() && stack.peek() < val){
            stackMin.push(stack.pop());
        }
        //放入栈中
        stack.push(val);
        //再借助辅助栈装回栈中,保证最小的永远在栈中
        while(!stackMin.isEmpty()){
            stack.push(stackMin.pop());
        }
    }

    public void pop() {
        //注意空处理
        if(isEmpty())
            return;
        stack.pop();
    }

    public int peek() {
        //空处理
        if (isEmpty()) {
            return -1;
        }
        return stack.peek();
    }

    public boolean isEmpty() {
        return stack.isEmpty();
    }
}

队列的最大值

图片说明

class MaxQueue {
    Deque<Integer> res;
    Deque<Integer> max;
    public MaxQueue() {
        res = new LinkedList<>();
        max = new LinkedList<>();
    }

    public int max_value() {
        if(max.isEmpty())
            return -1;
        //返回max的头部
        return max.peekFirst();
    }

    public void push_back(int value) {
        res.addLast(value);
        //更新max队列,保证是单调递减的
        while(!max.isEmpty() && max.peekLast() < value){
            max.removeLast();
        }
        max.addLast(value);
    }

    public int pop_front() {
        if(res.isEmpty())
            return -1;
        int temp = res.removeFirst();
        if(temp == max.peekFirst()){
            max.removeFirst();
        }
        return temp;
    }
}
字节算法题解 文章被收录于专栏

最近在刷字节的题库!! 春招给我冲!!!

全部评论

相关推荐

牛客279957775号:铁暗恋
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务