小红的中位数题解

这道题的关键在于判断删除该元素对中位数的影响,其实不难发现与删除的位置与原数组的长度有关,当为偶数时,删掉n/2以后的位置对原数组无影响,反之造成中位数后移一位。当原数组为奇数时,删掉n+1/2以后的位置造成数组前移,反之后移,注意特判原数组删中位数的情况:

#include<set>
#include<map>
#include<stack>
#include<queue>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
using ll=long long ;
using pii=pair<int,int> ;
using pdi=pair<double,int>;
#define endl "\n"

constexpr int N=1e5+10;
pdi p[N];
double a[N];
int pre[N],rear[N];

void solve()
{
    int n;
    cin>>n;
    
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        p[i]={a[i],i};
    }
    
    sort(p+1,p+1+n,[&](pdi a,pdi b)
         {
             return a.first<b.first;
         });
    
    for(int i=1;i<=n;i++)
    {
        int tag=p[i].second;
        pre[tag]=i;
    }
    
    if(n%2==0)
    {
        for(int i=1;i<=n;i++)
        {
            if(pre[i]>n/2)printf("%.1lf\n",p[n/2].first);
            else printf("%.1lf\n",p[n/2+1].first);
        }
    }
    else
    {
        for(int i=1;i<=n;i++)
        {
            if(pre[i]<(n+1)/2)printf("%.1lf\n",(p[(n+1)/2].first+p[(n+1)/2+1].first)/2.0);
            else if(pre[i]==(n+1)/2)printf("%.1lf\n",(p[(n)/2].first+p[(n+1)/2+1].first)/2.0);
            else printf("%.1lf\n",(p[(n)/2].first+p[(n+1)/2].first)/2.0);
        }
    }
}
signed main()
{
    //ios::sync_with_stdio(false);
    //cin.tie(0);cout.tie(0);
    
    int t;
    t=1;
    
    while(t--)
    solve();
    return 0;
}