2021年度训练联盟热身训练赛第一场-G Birthday Paradox ( 数学 )

题目链接：点击进入

题目

题意

m个人，n组生日（可能这一天就一个人过生日，也可能多个人过生日），一年365天，问给定的这种情况的概率

思路

答案概率 = 符合题意的情况数 / 总情况数
设 m 是总人数，那么总的情况数是 365 ^ m
符合题意的情况数：
1、选出 n 天生日的情况数:
2、 m 人分成 n 组的情况数（排除组内重复）

3、 n 个日期与 n 个组一一对应的情况数（因为每个组的人数可能相同，所以用记录每组人数出现的次数的数组 d 来计算，相当于 n 组分成 k 块（ k 是数组 d 的长度 ）），情况数为：

综上：
答案为

（ 可以预处理阶乘以 10 为底的对数 ）

代码

#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
//#define int long long
//#define double long double
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
//const double eps=1e-7;
const double pai=acos(-1.0);
const int N=2e4+10;
const int maxn=1e5+10;
const int mod=1e9+7;
const double eps=1e-12;
double lgfac[maxn],ans;
int n,a[maxn],cnt[maxn],sum,maxx;
int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
    for(int i=1;i<maxn;i++) 
		lgfac[i]=lgfac[i-1]+log10(i);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		sum+=a[i];
		cnt[a[i]]++;
		maxx=max(maxx,a[i]);
	}
	ans=lgfac[365]-lgfac[365-n]-lgfac[n]+lgfac[sum]+lgfac[n]-sum*log10(365);
    for(int i=1;i<=n;i++) 
		ans-=lgfac[a[i]];
    for(int i=1;i<=maxx;i++) 
		ans-=lgfac[cnt[i]];
    printf("%.15f\n",ans);
    return 0;
}