小红的中位数题解
这道题的关键在于判断删除该元素对中位数的影响,其实不难发现与删除的位置与原数组的长度有关,当为偶数时,删掉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;
}