Atcoder Beginner 057 D Maximum Average Sets 数学+思维

点击打开链接

题意:n件物品,可以选a~b件,求平均值最大是多少和方法数? n<=50,vi<=1e15
固定的k=a~b ave=sum/k ave要大 sum要最大 选最大的k个 
1/k*segma(vi) i=0~k-1 - 1/(k+1) segma(vi) i=0~k >=0
所以选A个时的平均值最大;


e.g a=2,b=4 a[i]={20,20,20,20,10} 方法数为11  
sum/k=(sum+x)/(k+1)  x=sum/k=ave, kx=sum=segma(a[i]) 因为从大->小排序,i<A,a[i]>=x a[i]只能都等于x
增加个数,平均值不变:只有当前A个相同 && 后面还有x=v[A-1]=..v[0] 方法数为num为v[0]的个数,ans+=C[num][A~B]
[x,x,y,y,z..z], x>y>z,无法增加个数,则方法数为 最小数z,共有ni个,在A中有ki个,则ans=C(ni,ki)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+20;
const int M=2e2+20;
ll v[N];
ll C[M][M]; 
void init()
{
	memset(C,0,sizeof(C));
	C[0][0]=1;
	for(int i=1;i<=50;i++)
	{
		C[i][0]=1;
		for(int j=1;j<=i;j++)
			C[i][j]=C[i-1][j-1]+C[i-1][j];
	}
}
int main()
{
	int n,A,B;
	init();
	while(cin>>n>>A>>B)
	{
		double ave=0;
		for(int i=0;i<n;i++)
			cin>>v[i];
		sort(v,v+n);
		reverse(v,v+n);//大->小
		for(int i=0;i<A;i++) 
			ave+=v[i];
		ave/=A;
		int num=0,k=0;
		for(int i=0;i<n;i++)
		{
			if(v[i]==v[A-1])
			{
				num++;
				if(i<A)
					k++;
			}
		}	
		ll ans=0; 
		if(k==A)//前A个相同,可以增加个数 
		{
			for(int i=A;i<=B;i++)
				ans+=C[num][i];
		}
		else
			ans+=C[num][k];

 		cout.precision(20);
 		cout<<fixed<<ave<<endl;
 		cout<<ans<<endl;
	}
	return 0;
}