洛谷P1168 中位数题解

最新推荐文章于 2022-07-16 10:54:39 发布

q779

最新推荐文章于 2022-07-16 10:54:39 发布

阅读量281

点赞数

分类专栏： OI 文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_50332374/article/details/125749006

版权

OI 专栏收录该内容

240 篇文章 3 订阅

订阅专栏

洛谷P1168 中位数题解

题目链接：P1168 中位数

题意：

给出一个长度为 $N$ 的非负整数序列 $A_i$ ，对于所有 $1 \leq k \leq (N + 1) / 2$ ，输出 $A_1, A_1 \sim A_3, …,A_1 \sim A_{2k - 1}$ 的中位数。即前 $1, 3, 5, \dots$ 个数的中位数。

对于 $100\%$ 的数据， $N \leq 100000$ 。

当然可以用vector来搞啦，但是这样就常数大~~而且没劲了~~

考虑对顶堆（那篇讲对顶堆的是在绕口令吧。明明很好懂的）

维护两个堆，分别是大根堆和小根堆

有个性质：大根堆q1里的所有值小于小根堆q2里的所有值

然后每次把比q1堆首元素大的元素扔到q2里去，反之塞进q1里

这样还是满足性质的

但是我们要中位数，这就需要q1和q2的size只相差一，才好处理

怎么搞呢，比如q1的size太大了，那我们就把q1的堆首一个个扔到q2里去

这样仍然是满足性质的

注意到每个元素最多入堆2次，因此单次插入操作是 $O(\log n)$ 的

时间复杂度 $O(n\log n)$

代码：

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iomanip>
#include <random>
#include <queue>
using namespace std;
#define int long long
#define INF 0x3f3f3f3f3f3f3f3f
#define N (int)()

int n;
priority_queue<int> q1;
priority_queue<int,vector<int>,greater<int>>q2;
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    // freopen("check.in","r",stdin);
    // freopen("check.out","w",stdout);
    cin >> n;
    int x; cin >> x;
    q1.push(x); cout << x << '\n';
    for(int i=2; i<=n; i++)
    {
        cin >> x;
        if(x>q1.top())q2.push(x);
        else q1.push(x);
        while(abs((int)(q1.size()-q2.size()))>1)
        {
            if(q1.size()>q2.size())
                q2.push(q1.top()),q1.pop();
            else q1.push(q2.top()),q2.pop();
        }
        if(i&1) cout << (q1.size()>q2.size()?q1.top():q2.top()) << '\n';
    }
    return 0;
}