码迷,mamicode.com
首页 > 移动开发 > 详细

[SRM478]RandomApple

时间:2018-09-04 14:01:52      阅读:175      评论:0      收藏:0      [点我收藏+]

标签:limits   rac   for   har   stdio.h   sum   highlight   length   mit   

题意:有$k$种苹果和$n$个箱子,每个箱子中有一些苹果,先等概率选取$n$个箱子组成集合的非空子集,再从选出的苹果中随机选一个,问每种苹果被选中的概率是多少

设箱子$i$有$cnt_{i,j}$个第$j$种苹果,第$i$个箱子的总苹果数$siz_i=\sum\limits_{j=1}^kcnt_{i,j}$,苹果总数$sum=\sum\limits_{i=1}^nsiz_i$

因为选完非空子集后,每个箱子中的苹果被选中的概率是相同的,所以先考虑选箱子,设$f_{i,j}$表示在前$i$个箱子中选$j$个苹果的方案数(只能一箱一箱选),那么$f_{i,j}=f_{i-1,j}+f_{i-1,j-siz_i}$

再设$g_{i,j}$表示在所有不是$i$的箱子中选$j$个苹果的方案数,那么$g_{i,j}=f_{n,j}-g_{i,j-siz_i}$

于是,选中了箱子$i$的情况对箱子$i$中每个苹果被选中的概率贡献为$\frac1{2^n-1}\sum\limits_{j=0}^{sum-siz_i}\frac{g_{i,j}}{j+siz_i}$(枚举箱子$i$外被选中的苹果数)

#include<stdio.h>
#include<vector>
#include<string>
#include<string.h>
using namespace std;
typedef double du;
typedef long long ll;
ll f[500010],g[500010];
int cnt[60][60],siz[60],sum;
class RandomApple{
	public:
		vector<du>theProbability(vector<string>hun,vector<string>ten,vector<string>one){
			int n,k,i,j;
			du t;
			n=hun.size();
			k=hun[0].length();
			for(i=0;i<n;i++){
				for(j=0;j<k;j++){
					cnt[i][j]=(hun[i][j]-‘0‘)*100+(ten[i][j]-‘0‘)*10+one[i][j]-‘0‘;
					siz[i]+=cnt[i][j];
				}
				sum+=siz[i];
			}
			vector<du>res(k,0);
			f[0]=1;
			for(i=0;i<n;i++){
				for(j=sum;j>=siz[i];j--)f[j]+=f[j-siz[i]];
			}
			for(i=0;i<n;i++){
				memset(g,0,sizeof(g));
				t=0;
				for(j=0;j<=sum;j++){
					g[j]=f[j]-(j>=siz[i]?g[j-siz[i]]:0);
					if(j+siz[i]<=sum)t+=g[j]/(du)((1ll<<n)-1)/(du)(j+siz[i]);
				}
				for(j=0;j<k;j++)res[j]+=t*cnt[i][j];
			}
			return res;
		}
};
/*
int main(){
	RandomApple cl;
	vector<string>a,b,c;
	vector<du>res;
	char s[60];
	for(scanf("%s",s);s[0]!=‘-‘;scanf("%s",s))a.push_back(s);
	for(scanf("%s",s);s[0]!=‘-‘;scanf("%s",s))b.push_back(s);
	for(scanf("%s",s);s[0]!=‘-‘;scanf("%s",s))c.push_back(s);
	res=cl.theProbability(a,b,c);
	for(du t:res)printf("%.9lf ",t);
}
*/

[SRM478]RandomApple

标签:limits   rac   for   har   stdio.h   sum   highlight   length   mit   

原文地址:https://www.cnblogs.com/jefflyy/p/9583530.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!