开锁魔法III | ||||||
| ||||||
Description | ||||||
一日,崔克茜来到小马镇表演魔法。 其中有一个节目是开锁咒:舞台上有 n 个盒子,每个盒子中有一把钥匙,对于每个盒子而言有且仅有一把钥匙能打开它。初始时,崔克茜将会随机地选择 k 个盒子用魔法将它们打开。崔克茜想知道最后所有盒子都被打开的概率,你能帮助她回答这个问题吗? | ||||||
Input | ||||||
第一行一个整数 T (T <= 100)表示数据组数。 每组数据第一行有两个整数 n 和 k (1 <= n <= 300, 0 <= k <= n)。 第二行有n个整数ai,表示第i个盒子中,装有可以打开第ai个盒子的钥匙。 | ||||||
Output | ||||||
对于每组数据,输出一行,包含一个实数,代表对应的答案。结果保留四位小数。 | ||||||
Sample Input | ||||||
4 5 1 2 5 4 3 1 5 2 2 5 4 3 1 5 3 2 5 4 3 1 5 4 2 5 4 3 1 | ||||||
Sample Output | ||||||
0.0000 0.6000 0.9000 1.0000 | ||||||
Source | ||||||
哈尔滨理工大学第五届ACM程序设计竞赛 |
思路:
①因为一个钥匙只能开一把门,所以如果我们开了一个门之后,接下来按照这把钥匙去继续开门直到不能开为止的话,他一定会构成一个环。所以我们可以O(n)暴力预处理出来环的信息,我们可以很容易得到环数m,以及第i个环包含的点数b【i】;
②那么对于这样一个求概率的问题,我们考虑到数据范围不是很大,我们可以设定Dp【i】【j】表示我们Dp过程到第i个环,一共选了j个点的方案数,我们只要保证每个环都至少选了一个点的话,那么dp出来的方案,就都是可行方案。
那么Ans=Dp【n】【k】/C(n,k);
状态转移方程也不难写出:Dp【i】【j】+=Dp【i-1】【j-l】*C(b【i】,l);
③问题难点在于,数据范围虽然不大,但是所有数据都会炸LL,没有进行结果最坏的估计,兴许还会炸double,所以我们不妨直接设定所有数据的类型都为long double即可,那么预处理出组合数,然后Dp写一写就行了。
Ac代码:
#include<stdio.h>
#include<string.h>
#include<iostream>
#include <iomanip>
using namespace std;
int a[500];
int vis[500];
int b[500];
long double c[500][500];
long double dp[500][500];
void GetGroup()
{
c[0][0] = c[1][0] = c[1][1] = 1;
for (int i=2; i<450; ++i)
{
c[i][0] = 1;
for (int j=1; j<=i; ++j)
c[i][j] = c[i-1][j] + c[i-1][j-1];
}
return ;
}
int main()
{
GetGroup();
int t;
cin>>t;
while(t--)
{
int m=0;
int n,k;
cin>>n>>k;
memset(vis,0,sizeof(vis));
for(int i=1;i<=n;i++)cin>>a[i];
for(int i=1;i<=n;i++)
{
if(vis[i]==0)
{
int len=0;
++m;
while(vis[a[i]]==0)
{
len++;
vis[a[i]]=1;
i=a[i];
}
b[m]=len;
}
}
memset(dp,0,sizeof(dp));
dp[0][0]=1;
for(int i=1;i<=m;i++)
{
for(int j=i;j<=k;j++)
{
for(int l=1;l<=b[i];l++)
{
if(j-l>=0)dp[i][j]+=dp[i-1][j-l]*c[b[i]][l];
}
}
}
cout<<fixed<< setprecision(4) << (dp[m][k]/c[n][k]) <<endl;
}
}