CodeForces 457 D.Bingo!（期望）

Description

用$m$个不同的数字去放到一个$n\times n$的棋盘中，然后随机从这$m$个数中取一个$k$子集，把棋盘中出现这$k$个数的位置标记，对于这个状态，其得分为$2$^{被完全标记的行数$+$被完全标记的列数}，问得分的期望值，取期望值与$10^{99}$的较小值输出

Input

三个整数$n,m,k(1\le n\le 300,n^2\le m\le 10^5,n\le k\le m)$

Output

输出得分期望值与$10^99$的较小值

Sample Input

1 2 1

Sample Output

2.5

Solution

一状态$sta$被标记行列数和为$f(sta)$，则其对分数总和贡献为$2^{f(sta)}$，而注意到$2^{f(sta)}$为从这些行列集中选出一子集方案数，设状态集为$S$，则得分期望值$ans=\frac{1}{|S|}\sum\limits_{sta\in S}2^{f(sta)}=\frac{1}{|S|}\sum\limits_{sta\in S}\sum\limits_{r=0}^n\sum\limits_{c=0}^n$对于状态$sta$被完全标记的行列中选出$r$行$c$列的方案数$=\sum\limits_{r=0}^n\sum\limits_{c=0}^nC_n^r\cdot C_n^c\cdot p_{r,c}$

其中$p_{r,c}$为选出的$r$行$c$列被标记的概率，不考虑数字之间的排列，总方案数为$C_m^k$，令$z=n\cdot (r+c)-r\cdot c$为这$r$行$c$列的格子数，则这$z$个位置必然在选出的$k$个数中，然后要从剩下的$m-z$个数中选出$k-z$个数字组成这$k$个数字，方案数$C_{m-z}^{k-z}$，故$p_{r,c}=\frac{C_{m-z}^{k-z}}{C_{m}^k}$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
ld f[maxn];
ld C(int n,int m)
{
    return f[n]-f[m]-f[n-m];
}
int main()
{
    f[0]=0;
    for(int i=1;i<=1e5;i++)f[i]=f[i-1]+log(1.0*i);
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    ld ans=0;
    for(int x=0;x<=n;x++)
        for(int y=0;y<=n;y++)
            if(n*(x+y)-x*y<=k)
            {
                int z=n*(x+y)-x*y;
                ld temp=C(n,x)+C(n,y)+C(m-z,k-z)-C(m,k);
                ans=min((ld)1e99,ans+exp(temp));
            }
    printf("%.10f\n",(double)ans);
    return 0;
}