大致题意
给n个数ai (n<=2e5,ai<=1e9,n是奇数), 可以执行k次操作,每次操作可以将其中一个数ai加1,求k次操作后中位数最大为多少。
思路
先排序,贪心的考虑mid=(n+1)>>1 之后的数据,前面的不用加。然后就是从mid一直到n,不断的将这段数字加到和后面第一个大于它的数字,相当与整体中位数增加了。(就像是一个阶梯,将mid之后的不断抬高到最后和最高一级一样)如果还有剩下的k ,加上k/(n-mid+1)即可。将整个过程模拟一下。O(n);
代码
#include<bits/stdc++.h>
using namespace std;
#define maxn 200005
#define maxm 1000006
#define ll long long int
#define INF 0x3f3f3f3f
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
#define mem(a) memset(a,0,sizeof(a))
#define sqr(x) (x*x)
#define inf (ll)2e18+1
int read(){
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
return f*x;
}
int n,k,a[maxn];
int main()
{
n=read();k=read();
inc(i,1,n)a[i]=read();
sort(a+1,a+n+1);
int mid=(n+1)/2;
ll ans=0;
inc(i,mid,n-1){
if(a[i]==a[i+1])continue;
else {
int len=i-mid+1;
int d=a[i+1]-a[i];
if(1ll*d*len>1ll*k){
int p=k/len;
ans+=p;
k=0;
break;
}
else{
ans+=d;
k-=d*len;
}
}
}
if(k)ans+=k/((n+1)/2);
printf("%lld\n",1ll*a[mid]+ans);
return 0;
}