PriorityQueue<Integer> A, B; public MedianFinder41() { //规定 小顶堆 保存较大的一半 长度为N/2 或 (N+1)/2 A = new PriorityQueue<>(); //规定 大顶堆 保存较小的一半 长度为N/2 或 (N-1)/2 B = new PriorityQueue<>((x, y) -> (y - x)); } public void addNum(int num) { //保持平衡 AB不能相差超过1 if(A.size() != B.size()) { // 条件成立 证明当前是奇数 再插入则为偶数了 // 为了平衡 A先进队 记住是有序的 A.add(num); // B进队 A出队 进出的是A里最小的 就平衡了 B.add(A.poll()); } else { // 不成立 即为当前偶数 在插入即为奇数 // 按照规则 这时候应该A比B多1 同理 B.add(num); A.add(B.poll()); } } public double findMedian() { // 条件成立 证明是奇数 从小顶堆取值 return A.size() != B.size() ? A.peek() : (A.peek() + B.peek()) / 2.0; }
剑指Offer-41数据流中的中位数
猜你喜欢
转载自blog.csdn.net/a792396951/article/details/113979049
今日推荐
周排行