标签：11 LeetCode295 int big 中位数 num 21.03 small public

中位数是有序列表中间的数。如果列表长度是偶数，中位数则是中间两个数的平均值。

例如，

[2,3,4] 的中位数是 3

[2,3] 的中位数是 (2 + 3) / 2 = 2.5

设计一个支持以下两种操作的数据结构：

void addNum(int num) - 从数据流中添加一个整数到数据结构中。
double findMedian() - 返回目前所有元素的中位数。
示例：

addNum(1)
addNum(2)
findMedian() -> 1.5
addNum(3)
findMedian() -> 2

class MedianFinder {

    /** initialize your data structure here. */
    private PriorityQueue<Integer> big;
    private PriorityQueue<Integer> small;
    public MedianFinder() {
        big = new PriorityQueue<>(new MaxHeapComparator());
        small = new PriorityQueue<>(new MinHeapComparator());
    }
    //先写大根堆和小根堆的排序规则
    public static class MaxHeapComparator implements Comparator<Integer> {
        public int compare(Integer o1,Integer o2)
        {
            if(o2>o1)
                return 1;
            else {
                return-1;
            }
        }
    }
    public static class MinHeapComparator implements Comparator<Integer> {
        public int compare(Integer o1,Integer o2) {
            if(o2<o1)
                return 1;
            else {
                return -1;
            }
        }
    }
    //调整大小根堆的大小
    public void modifyData()
    {
        //对比大小根堆的大小
        if(this.big.size() == this.small.size() +2 )
            this.small.add( this.big.poll() );
        if(this.small.size() == this.big.size() +2 )
            this.big.add(this.small.poll() );
    }
    public void addNum(int num) {
        if(this.big.isEmpty())
        {
            this.big.add(num);
            return;
        }
        if(this.big.peek()>=num)
        {
            this.big.add(num);
        }
        else{
            if(this.small.isEmpty())
            {
                small.add(num);
            }
            else{
                if(small.peek()>num)
                {
                    this.big.add(num);
                }
                else{this.small.add(num);}
            }
        }
        modifyData();
    }
    
    public double findMedian() {
        int maxSize = this.big.size();
        int minSize = this.small.size();
        if(maxSize+minSize==0)
            return 0;
        double maxMedian = this.big.isEmpty()? 0:this.big.peek();
        double minMedian = this.small.isEmpty()?0:this.small.peek() ;
        double res = (maxMedian+minMedian)/2;
        if ( (maxSize+minSize) %2!=1 )//序列是否为偶数
            return res;
        return maxSize>minSize?maxMedian:minMedian;
    }
}

标签：11,LeetCode295,int,big,中位数,num,21.03,small,public
来源： https://www.cnblogs.com/pionice/p/14516527.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。