题目链接
题意:找最大的中位数,可以操作k次,每次可以给某个数+1;
思路:从中位数到最后一次遍历,相等就下一个,比前面的大就对前面的数(从当前到中位数)+1,知道数组最后或者k<=0;
个人思考:本人思路没错,但在实施的时候不知道出了什么问题一直WA 6,有点郁闷,最后还是没忍住去偷看了大佬的代码;
根据大佬的思路写的代码(基本就是抄的):
#include <iostream>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <set>
#include <list>
#include <queue>
#include <stack>
#include <vector>
#include <algorithm>
#define ULL unsigned long long
using namespace std;
const int maxn=2e5+10;
long long int a[maxn];
int main()
{
long long int i,n,m,j,k,l,p,q,s;
scanf("%lld%lld",&n,&k);
for(i=0; i<=n-1; i++)
{
scanf("%lld",&a[i]);
}
sort(a,a+n);
long long int cnt=1;
long long int ans=a[n/2];
for(i=n/2;k>0&&i+1<n;i++)
{
long long int v=min(a[i+1]-ans,k/cnt);//选择其中较小的,如果后者较小,k=0,跳出循环
ans=ans+v;
k=k-v*cnt;//余额减少
cnt++;//区间+1
}
if(k>0)
{
ans=ans+(k/cnt);
}
printf("%lld\n",ans);
return 0;
}
自己写的代码WA 6
#include <iostream>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <set>
#include <list>
#include <queue>
#include <stack>
#include <vector>
#include <algorithm>
#define ULL unsigned long long
using namespace std;
const int maxn=2e5+10;
long long int a[maxn];
int main()
{
long long int i,n,m,j,k,l,p,q,s;
while(~scanf("%lld%lld",&n,&k))
{
for(i=0; i<=n-1; i++)
{
scanf("%lld",&a[i]);
}
sort(a,a+n);
while(k>0)
{
s=1;
for(i=n/2; i<=n-1; i++)
{
//printf("1\n");
if(a[i]==a[i+1])
{
s++;
}
else
{
p=a[i+1]-a[i];
break;
}
}
//printf("%d\n",s);
if(s==n-n/2)
{
for(p=1;p*s<=k;p++)
{
//printf("yes\n");
}
}
if(p*s<=k)
{
k=k-p*s;
for(i=n/2; i<=n/2+s; i++)
{
a[i]=a[i]+p;
}
//printf("%lld\n",p);
}
else
{
//printf("9\n");
q=p-1;
for(; q>0; q--)
{
if(q*s<=k)
{
k=k-q*s;
for(i=n/2; i<=n/2+s; i++)
{
a[i]=a[i]+q;
}
break;
}
}
if(q<=0)
{k=0;}
}
}
printf("%lld",a[n/2]);
return 0;
}
}