bzoj5004 开锁魔法II dp

Description

---

一日，崔克茜来到小马镇表演魔法。其中一个节目是开锁咒：舞台上有n
个盒子，每个盒子中有一把钥匙，对于每个盒子而言有且仅有一把钥匙能打开它。
初始时，崔克茜将会随机地选择k个盒子用魔法将它们打开。崔克茜想知道最后
所有盒子都被打开的概率，你能帮助她回答这个问题吗？

Solution

---

考虑钥匙向盒子连边，形成的图一定为多个不相交的环，我们只需要一个环至少选一个就能打开环内剩余的盒子
注意到直接求不太好做，我们dp设f[i,j]为前i个环用了j个钥匙都打开了的方案数，我们用答案除总的方案数就可以了
这个转移比较显然，组合数直接用double不会有毛病

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))

typedef long long LL;
const int N=505;

struct edge {int y,next;} e[N*2];

double c[N][N],f[N][N];

int ls[N],size[N],wjp,edCnt;

bool vis[N];

void add_edge(int x,int y) {
   e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
}

void dfs(int now) {
   vis[now]=true; wjp++;
   for (int i=ls[now];i;i=e[i].next) {
   	if (vis[e[i].y]) continue;
   	dfs(e[i].y);
   }
}

int main(void) {
   freopen("data.in","r",stdin);
   freopen("myp.out","w",stdout);
   c[1][0]=c[1][1]=1;
   rep(i,2,N-1) {
   	c[i][0]=1;
   	rep(j,1,i) c[i][j]=c[i-1][j]+c[i-1][j-1];
   }
   int T; for (scanf("%d",&T);T--;) {
   	fill(f,0); fill(ls,0); edCnt=0;
   	int n,m,tot=0; scanf("%d%d",&n,&m);
   	rep(i,1,n) {
   		int x; scanf("%d",&x);
   		add_edge(x,i); vis[i]=false;
   	}
   	rep(i,1,n) if (!vis[i]) {
   		wjp=0; dfs(i);
   		size[++tot]=wjp;
   	}
   	rep(i,1,std:: min(m,size[1])) {
   		f[1][i]=c[size[1]][i];
   	}
   	rep(i,2,tot) rep(j,1,m) {
   		rep(k,1,j-1) {
   			f[i][j]+=f[i-1][j-k]*c[size[i]][k];
   		}
   	}
   	printf("%.9lf\n", 1.0*f[tot][m]/c[n][m]);
   }
   return 0;
}