最近在准备找工作的过程中刷了很多的题,也看了很多的技术性文章,但是有很多当时记忆犹新的知识点以及一些心得慢慢的已经开始遗忘。因此想要从今天开始写一些博客,记录一下这个让人又爱又恨的做题之旅。(还可以锻炼一下自己的语言表达能力)
题目描述
输入一个整型数组,数组里有正数也有负数。数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为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;
}
}