码迷,mamicode.com
首页 > 其他好文 > 详细

队列实现max操作,要求尽量提高效率。

时间:2014-08-20 01:18:46      阅读:229      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   java   ar   div   log   new   

结合之前实现的 maxStack 和 用两个stack 实现一个Queue, 实现 MaxQueue

 

import java.util.Stack;

public class MaxQueue {
    MaxStack in = new MaxStack();
    MaxStack out = new MaxStack();

    public void enqueue(Integer a) {
        in.push(a);
    }

    public Integer dequeue() {
        shiftItems();
        return out.pop();
    }

    public Integer peek() {
        shiftItems();
        return out.peek();
    }

    public Integer max() {
        return Math.max(in.isEmpty() ? Integer.MIN_VALUE : in.max(), out.isEmpty() ? Integer.MIN_VALUE : out.max());
    }

    private void shiftItems() {
        if (!out.isEmpty())
            return;
        while (!in.isEmpty()) {
            out.push(in.pop());
        }
    }

    public static void main(String[] args) {
        MaxQueue queue = new MaxQueue();
        queue.enqueue(1);
        System.out.println(queue.max());
        queue.enqueue(2);
        System.out.println(queue.max());
        queue.enqueue(3);
        System.out.println(queue.max());
        queue.enqueue(4);
        System.out.println(queue.max());
        queue.dequeue();
        System.out.println(queue.max());
        queue.dequeue();
        System.out.println(queue.max());
        queue.dequeue();
        System.out.println(queue.max());

    }
}

class MaxStack {
    Stack<Integer> val = new Stack<Integer>();
    Stack<Integer> max = new Stack<Integer>();

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

    public void push(Integer a) {
        val.push(a);
        if (max.isEmpty() || a >= max.peek())
            max.push(a);
    }

    public Integer pop() {
        Integer out = val.pop();
        if (out == max.peek())
            max.pop();
        return out;
    }

    public Integer peek() {
        return val.peek();
    }

    public Integer max() {
        return max.peek();
    }
}

 

队列实现max操作,要求尽量提高效率。,布布扣,bubuko.com

队列实现max操作,要求尽量提高效率。

标签:style   blog   color   java   ar   div   log   new   

原文地址:http://www.cnblogs.com/jdflyfly/p/3923631.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!