题目链接:【CodeForces 626E】
从长度为n的数列中找出一个子序列,使得这个子序列的(平均数-中位数)最大,输出这个子序列
先将原序列从小到大排序,再枚举中位数,求每一个中位数的最大平均数
子序列的长度一定是奇数,当个数由奇数个变成偶数个时,中位数增加的比平均数增加的要多,(平均数-中位数)的值就会变小
针对每一个中位数,二分总长度的长度的一半len,找到最大的平均数。
中位数是arr[i],长度的一半为len时,为了使平均数尽可能的大,找到的子序列一定是:arr[i-len]……arr[i] && arr[n-len+1]……arr[n](arr[]从1开始)
当len+1时,子序列中将增加arr[i-len-1]和arr[n-len],增加的数是减小的,所以随着len的增加,平均数总的来说应该是先增大再减小的,所以可以用二分求len
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
using namespace std;
#define ll __int64
const int inf=1e5+10;
int n;
ll arr[2*inf], sum[2*inf];
int main()
{
cin>>n;
sum[0]=0;
for(int i=1; i<=n; i++)
{
cin>>arr[i];
}
sort(arr+1, arr+n+1);
double ans=-1e9;
int len=0, it=1;
for(int i=1; i<=n; i++) sum[i] = sum[i-1]+arr[i];
for(int i=1; i<=n; i++)
{
int l=1, r=min(i-1, n-i), p=0;
while(l<=r)
{
int mid=(l+r)>>1;
int mi = mid-1;
double mean1=1.0*(sum[n]-sum[n-mid] + sum[i]-sum[i-mid-1])/(2*mid+1);
double mean2=1.0*(sum[n]-sum[n-mi] + sum[i]-sum[i-mi-1])/(2*mi+1);
if(mean1<mean2) r=mid-1;
else l=mid+1, p=mid;
}
double tmp=1.0*(sum[n]-sum[n-p] + sum[i]-sum[i-p-1])/(2*p+1)-arr[i];
if(ans<tmp)
{
ans = tmp;
len = p, it = i;
}
}
cout<<len*2+1<<endl;
for(int i=it-len; i<=it; i++) cout<<arr[i]<<" ";
for(int i=n-len+1; i<=n; i++) cout<<arr[i]<<" ";
cout<<endl;
return 0;
}