[2019寒假集训day2]子集(二分)

题面

题解

签到题。
首先可以证明的是，选出的子集大小一定是奇数个。
考虑先枚举中位数，因此可以排一下序。
设选的中位数位置为$p$,那么我们需要在$p$前面和后面分别选出$k$个，并最大化平均值。
显然，选出的这$2k$个数越靠近右边越好。
这样，每次该加入子集的两个数就是确定的。
复杂度$O(n^2)$
注意到$p$左右两边的序列都有单调性。
如果我们发现位置$p$的最优解也有单调性，那就可以愉快的二分了。
考虑代数证明一下。
设现在已经在$p$左右两边分别选出了$k-1$个数，现在考虑再往两边分别加入1个数，平均值没有之前优时满足的情况。
设左边第$i$次选出数$A_i$,右边第$i$次选出数$B_i$
那么有:
$$\frac{{\sum }_{i=1}^{k-1}{A}_i+a_p+{\sum }_{i=1}^{k-1}{B}_i}{2k-1}\ge \frac{{\sum }_{i=1}^k{A}_i+a_p+{\sum }_{i=1}^k{B}_i}{2k+1}$$
化简:
$$(2k+1)(\sum _{i=1}^{k-1}{A}_i+a_p+\sum _{i=1}^{k-1}{B}_i)\ge (2k-1)(\sum _{i=1}^k{A}_i+a_p+\sum _{i=1}^k{B}_i)$$
$$2(\sum _{i=1}^{k-1}{A}_i+a_p+\sum _{i=1}^{k-1}{B}_i)\ge (2k-1)(A_k+B_k)$$
$$\frac{{\sum }_{i=1}^{k-1}{A}_i+a_p+{\sum }_{i=1}^{k-1}{B}_i}{2k-1}\ge \frac{A_k+B_k}{2}$$
其实就是新加入的平均值小于原来的平均值就不优。
可以发现$A_k+B_k$是递减的，数列的平均值是单峰的，符合二分条件。
复杂度$O(n\log n)$

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define MAXN 200000
#define DB double
#define LL long long
int n,a[MAXN+5];
DB sum[MAXN+5];
DB ans;
int main()
{
	//freopen("subset.in","r",stdin);
	//freopen("subset.out","w",stdout);
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	sort(a+1,a+n+1);
	for(int i=1;i<=n;i++)sum[i]=sum[i-1]+a[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;
			DB k1=(DB)(sum[i]-sum[i-mid]+sum[n]-sum[n-mid+1])/(2*mid-1);
			DB k2=(DB)(a[i-mid]+a[n-mid+1])/2.0;
			if(k1<=k2)p=mid,l=mid+1;
			else r=mid-1;
		}
		DB res=(DB)(sum[i]-sum[i-p-1]+sum[n]-sum[n-p])/(DB)(2*p+1);
		ans=max(ans,res-a[i]);
	}
	printf("%0.5lf\n",ans);
}