题意
n个房间的钥匙随机分布在这n个房间中,每个房间一个钥匙。每次先破环一个门,拿到一个钥匙,如果这个钥匙可以打开其他没开的门,就用它打开下一个门,取得下一个钥匙,如此重复,直到拿到的钥匙打不开其他没开的门为止,这时再破坏一个门,重复前面的操作。可以最多破坏掉k个门。注意第一个门不能被破坏,只能由钥匙打开。给出n和k,求能打开全部门的概率。
思路
首先,设所求概率为p,则
其中s'(n,i)表示破坏i(1<=i<=k)个门得以解决问题的钥匙分配方案数。
能用一把钥匙打开的x个门构成一个环。破坏环上任一个门之后,环内的所有房间都可以用钥匙进入。也就是说,题意可以理解成求n个房间形成1~k个环的总概率有多大。
第一类斯特林数表示将 n 个不同元素构成k个圆环排列的数目,正好可以用来解决此类问题。
递推公式:
S(n,k)= S(n-1,k-1)+(n-1)* S(n-1,k),其中1<=k<=n,S(1,1)=1,S(i,0)= 0 (1<=i<=n).
注意,由于第一个门不能被破坏,意味着它只能由钥匙打开,
所以要减去第一个门独立成环(即剩余n-1个房间组成k-1个环)的方案数,即s'(n,i)= S(n,k)- S(n-1,k-1)
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
typedef long long LL;
using namespace std;
const int N = 21;
LL fac[N],stir[N][N]; //阶乘和斯特林数已经超过int范围
int n,k;
void cal(){ //阶乘和斯特林数打表
fac[1]=1;
for(int i=2;i<N;i++)
fac[i]=i*fac[i-1];
stir[1][1]=1;
for(int i=2;i<N;i++){
for(int j=1;j<=i;j++){
stir[i][j]=stir[i-1][j-1]+(i-1)*stir[i-1][j];
}
}
}
int main(){
cal();
int T;
cin>>T;
while(T--){
LL sum=0;
scanf("%d%d",&n,&k);
for(int i=1;i<=k;i++)
sum+=(stir[n][i]-stir[n-1][i-1]);
printf("%.4lf\n",1.0*sum/fac[n]);
}
return 0;
}
第二类斯特林数表示将n个不同的元素拆分成k个非空集合的方案数,实际上是一个集合的划分。
和第一类Stirling数不同的是,集合内是不考虑次序的,而圆环排列是有序的。
常常用于解决组合数学中几类放球模型。描述为:将n个不同的球放入k个无差别的盒子中,要求盒子非空,有几种方案?
递推公式为:
S(n,k)= S(n-1,k-1)+ k * S(n-1,k),其中1<=k<=n,S(1,1)=1,S(i,0)= 0 (1<=i<=n).