#3124. 开锁(unlock)

题目描述

A君有 $n$ 个盒子，每个盒子被一把锁锁着，每个盒子内都有一把钥匙。对于每个盒子而言有且仅有一把钥匙能打开锁着它的锁，而打开它后便能拿着放置在这个盒子内的钥匙去开启其他盒子。

现在A君打算随机选择 $k$ 个盒子并用魔法将它们打开，并用所得到的钥匙去尝试开启其他所有的盒子 $f$ 开启一个盒子后，新得到的钥匙还能继续尝试使用）。

A君想知道，最终他能打开所有盒子的概率是多少，请你帮助他。

数据范围

$k\le n\le 300,T\le 100$

题解

考虑到只要每个置换圈被取出其中一个，那就可以打开所有盒子

所以设置换数为 $m$，第 $i$ 个置换中的个数为 $s_i$ ，考虑 $dp$ ，设 $f_{i,j}$ 表示前 $i$ 个，取了 $j$ 个的方案数，则$f_{i,j}={\sum }_{l=1}^{min(j,s_i)}{f}_{i-1,j-l}\times C_{s_i}^l$

则答案为 $\frac{f_{m,k}}{C_n^k}$

效率： $O(Tnk)$

代码

#include <bits/stdc++.h>
#define db long double
using namespace std;
db c[305][305],ans;
int T,k,a[305],m,b,n,s[305],f[305];
db g[305][305];
int main(){
	c[0][0]=1;
	for (int i=1;i<=300;i++){
		c[i][0]=1;
		for (int j=1;j<=i;j++)
			c[i][j]=c[i-1][j]+c[i-1][j-1];
	}
	for (scanf("%d",&T);T--;){
		scanf("%d%d",&n,&k);m=0;b=n;
		for (int i=1;i<=n;i++)
			scanf("%d",&a[i]),f[i]=0;
		for (int j,i=1;i<=n;i++)
			if (!f[i]){
				j=a[i];s[f[i]=++m]=1;
				while(!f[j])
					f[j]=m,j=a[j],s[m]++;
			}
		g[0][0]=1;
		for (int i=1;i<=m;i++)
			for (int j=1;j<=k;j++){
				g[i][j]=0;
				for (int l=1;l<=j && l<=s[i];l++)
					g[i][j]+=g[i-1][j-l]*c[s[i]][l];
			}
		ans=g[m][k];
		for (int i=1;i<=k;i++) ans/=n,n--,ans*=i;
		printf("%.10LF\n",ans);
	}
	return 0;
}