剑指offer-第41题-数据流中的中位数(java)

剑指offer-第41题-数据流中的中位数(java)

最近在准备找工作的过程中刷了很多的题,也看了很多的技术性文章,但是有很多当时记忆犹新的知识点以及一些心得慢慢的已经开始遗忘。因此想要从今天开始写一些博客,记录一下这个让人又爱又恨的做题之旅。(还可以锻炼一下自己的语言表达能力)

题目描述

输入一个整型数组,数组里有正数也有负数。数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。

要求时间复杂度为O(n)。
示例1:

输入: nums = [-2,1,-3,4,-1,2,1,-5,4]
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。

解题思路

这道题主要是要维护两个优先队列,一个升序,一个降序。要保持两个队列得以平衡,因为中位数是中间的那个数。队列维护大元素,B队列维护小元素。A在奇数个元素的时候比B多一个元素,在偶数个元素的时候和B相同。那在加入元素的时候如何维护这个平衡的状态呢?
在这里插入图片描述

用A、B队列元素个数相比来控制:如果A.size() ! = B.size(),则此时A比B多一个元素,应该往B里面加元素,注意不是直接把输入的num加进去,而是先把num加入A,再把A的队首元素弹出给B,这样保证B中都是小元素。同理,如果A.size() == B.size(),则此时A和B元素相等,往A里面添加元素,保证A比B多一个或相等,保持平衡。
队列的初始化是实现Queue接口:

class MedianFinder{
	Queue<Integer> A,B;
	public MedianFinder() {
	//优先队列自动排序,默认为升序,队头最小。
        A = new PriorityQueue<>();
    //再创建一个降序的优先队列,队头最大。
        B = new PriorityQueue<>((x,y)->(y-x));
    }
    
    public void addNum(int num) {
        if(A.size()!=B.size()){
            A.add(num);
            B.add(A.poll());
        }
        else{
            B.add(num);
            A.add(B.poll());
        }
    }
    
    public double findMedian() {
    //因为是返回double类型,所以是除以2.0
        return A.size()!=B.size() ? A.peek() : (A.peek()+B.peek())/2.0;
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值