P1168 中位数

Lsxp1991

于 2023-08-01 19:30:29 发布

阅读量133

点赞数

文章标签：算法 c++ 开发语言数据结构 c语言

本文链接：https://blog.csdn.net/m0_56631155/article/details/132049261

版权

中位数

题目描述

给定一个长度为 $N$ 的非负整数序列 $A$ ，对于前奇数项求中位数。

输入格式

第一行一个正整数 $N$ 。

第二行 $N$ 个正整数 $A_{1\dots N}$ 。

输出格式

共 $\lfloor \frac{N + 1}2\rfloor$ 行，第 $i$ 行为 $A_{1\dots 2i - 1}$ 的中位数。

样例 #1

样例输入 #1

7
1 3 5 7 9 11 6

样例输出 #1

提示

对于 $20\%$ 的数据， $\le 100$ ；

对于 $40\%$ 的数据， $\le 3000$ ；

对于 $100\%$ 的数据， $\le N ≤ 100000$ ， $\le A_i \le 10^9$ 。

分析

创建两个堆分别来放中位数两边的数，q1大根堆降序放小的数字，q2小根堆升序放大于等于的数字。
中位数放在q2队列的top，所以q2.size()应该比q1.size()大1，
如果不满足则弹出放入另一个队列，弹出的数为更新的mid值。

//方法一，构造两个堆
#include <iostream>
#include <queue>
using namespace std;
int main()
{
    priority_queue<int, vector<int>, less<int>> q1;    // 大根堆
    priority_queue<int, vector<int>, greater<int>> q2; // 小根堆
    int n;
    cin >> n;
    int mid;
    int a;
    cin >> a;
    mid = a;
    cout << a << endl;//预处理第一项，第一项一定会输出
    q2.push(a);//中位数放在q2队列的top，所以q2.size()应该比q1.size()大1;
    // q1+q2,将mid存入q2，输出q2.top
    // q1大根堆降序放小的数字，q2小根堆升序放大于等于的数字
    for (int i = 2; i <= n; i++)
    {
        cin >> a;
        if (a < mid)
            q1.push(a);
        else
            q2.push(a);
        if (i % 2 == 1)
        {
            while(q1.size() < q2.size())
            {
                mid = q2.top();
                q2.pop();
                q1.push(mid);
            }
            while(q1.size() > q2.size()){
                mid = q1.top();
                q1.pop();
                q2.push(mid);
            }
            cout << mid << endl;
        }
    }
}

还有另一种方法。
数据过大，sort等排序会超时，所以，可以每次加入新数的时候就插入在使数组有序的位置。
定义vector数组，方便新数的插入。

//方法二，vector配合upper_bound实现有序数组
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main()
{
    int n;
    vector<int>a;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        int x;
        cin>>x;
        a.insert(upper_bound(a.begin(),a.end(),x),x);
        if(i%2==1) cout<<a[(i-1)/2]<<endl;
    }
}