P1168 中位数——离散化+树状数组+二分

发布时间:2024年01月19日

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

输入
第一行一个正整数 N(1 ≤ N ≤ 1e5)。
第二行 N 个正整数 A1,……,AN(0≤ Ai≤ 1e9)。

输出格式
共 (N+1)/2 行,第 i 行为 A[1]到A[2i-1] 的中位数。

Input1
7
1 3 5 7 9 11 6
Output1
1
3
5
6

Input2
7
3 1 5 9 8 7 6

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
typedef pair<int,int> PII;
const double PI=acos(-1.0);
const int N=2e6+10;
int n;
int tr[N];
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int c)
{
    for (int i=x;i<=n;i +=lowbit(i)) tr[i] +=c;
}
int query(int x)
{
    int ans=0;
    for (int i=x;i>0;i -=lowbit(i)) ans +=tr[i];
    return ans;
}
int a[N];
map <int,int> s,p;
void solve()
{
    cin>>n;
    for (int i=1;i<=n;i++)
    {
        cin>>a[i];
        s[a[i]]++;
    }
    int cnt=1;
    for (auto &[x,y]:s)
    {
        p[cnt]=x;
        y=cnt;
        cnt++;
    }
    add(s[a[1]],1);
    int l=1,r=n;
    while (l<r)
    {
        int mid=l+r>>1;
        if (query(mid)>=1) r=mid;
        else l=mid+1;
    }
    cout<<p[l]<<endl;
    if (n%2==0) n--;
    for (int i=2;i<=n;i +=2)
    {
        add(s[a[i]],1),add(s[a[i+1]],1);
        int l=1,r=n;
        int m=i+2;
        while (l<r)
        {
            int mid=l+r>>1;
            if (query(mid)>=(m+1) /2) r=mid;
            else l=mid+1;
        }
        cout<<p[l]<<endl;
    }
}
signed main()
{
    ios;
    int T=1;
    //cin>>T;
    while (T--) solve();
    return 0;
}

Output2
3
3
5
6

文章来源:https://blog.csdn.net/m0_74403543/article/details/135689194
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。