Hrbust 2249开锁魔法II(dp)

开锁魔法II

Time Limit: 3000 MS Memory Limit: 256000 K

Total Submit: 41(32 users) Total Accepted: 31(28 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
题解:dp[i][j]表示I个锁j个魔法能打开的概率，则状态转移方程为：
dp[i][j]=(dp[i-1][j-1]+d[i-1][j]*(i-1))/i：开一个箱子，里面钥匙恰好是该箱子的（概率为1/i）+不是该箱子的（概率为（i-1)/i);
代码：

#include <stdio.h>
#include <iostream>
#include <stdlib.h>
#include <map>
#define bababaa printf("!!!!!!!\n")
using namespace std;
const int N=6000;
double dp[N][N];
void init()
{
    for(int i=1;i<=5000;i++)
    {
        dp[i][0]=0;
        dp[i][i]=1.0;
        for(int j=1;j<i;j++)
        {
            dp[i][j]=(dp[i-1][j-1]+dp[i-1][j]*(i-1))*1.0/i;
        }
    }
}
int main()
{
    init();
    int n,k;
    while(cin>>n>>k)
    {
        printf("%.4lf\n",dp[n][k]);
    }
    return 0;
}