hrbust 2249 开锁魔法II【概率dp】

开锁魔法II

Time Limit: 3000 MS Memory Limit: 256000 K Total Submit: 42(33 users) Total Accepted: 32(29 users) Rating:  Special Judge: No

Description

一日，崔克茜来到小马镇表演魔法。 其中有一个节目是开锁咒：舞台上有 n 个盒子，每个盒子中有一把钥匙，对于每个盒子而言有且仅有一把钥匙能打开它。崔克茜可以通过魔法，暴力打开一个盒子。但是崔克茜最多只可以使用 k 次魔法，问在 k 次魔法内，打开所有盒子的概率是多少。

Input

多组测试数据。 每组数据第一行有一个整数 n 和一个整数 k。(1 <= k <= n <= 5000)

Output

对于每组数据，输出一行表示对应的答案。结果保留四位小数。

Sample Input

3 1 3 2 3 3 10 3

Sample Output

0.3333 0.8333 1.0000 0.7061

Source

哈尔滨理工大学第五届ACM程序设计竞赛

思路:

1、考虑dp，设定dp【i】【j】表示开了i个锁，用了j次魔法将其全部打开了的概率。

2、那么考虑状态转移方程：
①首先考虑情况数，我们若从i-1的情况转移过来，一共可能会有i种情况，其中i-1种情况是从i-1个锁中任意一个锁打开后可以打开第i个锁，其中还有1种情况是表示第i个锁的钥匙在第i个盒子里边，只能用魔法打开的情况。

②那么不难推出其状态转移方程：

dp【i】【j】=（dp【i-1】【j-1】*1）/i，表示当前第i个盒子必须用一次魔法来打开的情况，也就是只有这一种情况。

dp【i】【j】+=（dp【i-1】【j】*（i-1））/i，表示当前第i个盒子不用魔法来打开的情况，其一共会有（i-1）种情况，表示从前i-1个盒子中，任意一个盒子里边的钥匙是第i个盒子的钥匙的情况。

Ac代码：

#include<stdio.h>
#include<string.h>
using namespace std;
double dp[5005][5005];
int main()
{
    for(int i=1;i<=5000;i++)
    {
        dp[i][0]=0;
        dp[i][i]=1;
        for(int j=1;j<i;j++)
        {
            dp[i][j]=(dp[i-1][j-1])/i+(dp[i-1][j]*(i-1))/i;
        }
    }
    int n,k;
    while(~scanf("%d%d",&n,&k))
    {
        printf("%.4lf\n",dp[n][k]);
    }
}