HRBUST 2250

题目链接：http://acm.hrbust.edu.cn/index.php?m=ProblemSet&a=showProblem&problem_id=2250

题意：随机从n个盒子中取k个钥匙，求打开全部盒子的概率。

首先用置换群的思想去将n个盒子，分成几组（具体情况看给出的数据），题意的数据可分为（2,5,1） （3,4）两组。

意思是只要拿到每组中的一个钥匙，即可开启该组的全部盒子。所以对于我们分成的组，每组至少拿到一个钥匙。

假设分为s组，则题意转化为对于k个钥匙，要在s组中取完，每组至少取一个。得到的方案数除以C（n，k）.

一开始想到是递归求解方案数，结果超时orz。。。看了别人的题解才发现是简单的计算DP即可求解出来（智商不够用）.

代码实现：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <vector>
#include <cmath>
using namespace std;

int n, k;

int perm[500];
bool used[500];

typedef long double ld;

ld comb[500][500];

ld dp[500][500] = {};

int main() 
{
	// freopen("in.txt","r",stdin);

	//组合数
	for (int i = 0; i<= 300; ++i) 
	{
		for (int j = 0; j <= i; ++j) 
		{
			comb[i][j] =(i == 0 || j == 0) ?
						1 : (comb[i - 1][j] + comb[i - 1][j - 1]);
		}
	}

	int T;
	scanf("%d",&T);
	while (T--) 
	{
		scanf("%d %d",&n,&k);
		vector<int> cyc;

		for (int i = 0; i < n; ++i) 
		{
			scanf("%d",&perm[i]);
			perm[i]--;
		}
		//置换群思想
		memset(used, false, sizeof(used));
		for (int i = 0; i < n; ++i) 
		{
			if (used[i])
				continue;
			int x = i;
			int cur = 0;
			while (!used[x]) 
			{
				++cur;
				used[x] = true;
				x = perm[x];
			}
			cyc.push_back(cur);
		}

		for (int i = 0; i <= 300; ++i) 
		{
			for (int j = 0; j <= 300; ++j) 
			{
				dp[i][j] = 0;
			}
		}

		//dp[i][j]：表示当前已经处理了前 i 组循环，
		//一共分配了 j 个初始钥匙时的概率即可。
		dp[0][0] = 1;
		for (int i = 0; i < cyc.size(); ++i) 
		{
			int x = cyc[i];
			for (int j = 0; j <= k; ++j) 
			{
				for (int take = 1; take + j <= k && take <= x; ++take) 
				{
					dp[i + 1][j + take] += dp[i][j] * comb[x][take];
				}
			}
		}

		ld ans = dp[cyc.size()][k] / comb[n][k];

		printf("%.4lf\n", (double) ans);
	}
}