HDU 3625 Examining the Rooms(第一类stirling数)

Problem Description
A murder happened in the hotel. As the best detective in the town, you should examine all the N rooms of the hotel immediately. However, all the doors of the rooms are locked, and the keys are just locked in the rooms, what a trap! You know that there is exactly one key in each room, and all the possible distributions are of equal possibility. For example, if N = 3, there are 6 possible distributions, the possibility of each is 1/6. For convenience, we number the rooms from 1 to N, and the key for Room 1 is numbered Key 1, the key for Room 2 is Key 2, etc.
To examine all the rooms, you have to destroy some doors by force. But you don’t want to destroy too many, so you take the following strategy: At first, you have no keys in hand, so you randomly destroy a locked door, get into the room, examine it and fetch the key in it. Then maybe you can open another room with the new key, examine it and get the second key. Repeat this until you can’t open any new rooms. If there are still rooms un-examined, you have to randomly pick another unopened door to destroy by force, then repeat the procedure above, until all the rooms are examined.
Now you are only allowed to destroy at most K doors by force. What’s more, there lives a Very Important Person in Room 1. You are not allowed to destroy the doors of Room 1, that is, the only way to examine Room 1 is opening it with the corresponding key. You want to know what is the possibility of that you can examine all the rooms finally.
 

Input
The first line of the input contains an integer T (T ≤ 200), indicating the number of test cases. Then T cases follow. Each case contains a line with two numbers N and K. (1 < N ≤ 20, 1 ≤ K < N)
 

Output
Output one line for each case, indicating the corresponding possibility. Four digits after decimal point are preserved by rounding.
 

Sample Input
 
   
3 3 1 3 2 4 2
 

Sample Output
 
   
0.3333 0.6667 0.6250
Hint
Sample Explanation When N = 3, there are 6 possible distributions of keys: Room 1 Room 2 Room 3 Destroy Times #1 Key 1 Key 2 Key 3 Impossible #2 Key 1 Key 3 Key 2 Impossible #3 Key 2 Key 1 Key 3 Two #4 Key 3 Key 2 Key 1 Two #5 Key 2 Key 3 Key 1 One #6 Key 3 Key 1 Key 2 One In the first two distributions, because Key 1 is locked in Room 1 itself and you can’t destroy Room 1, it is impossible to open Room 1. In the third and forth distributions, you have to destroy Room 2 and 3 both. In the last two distributions, you only need to destroy one of Room 2 or Room
 

题目大意:n 个房间,n 把钥匙,不同的钥匙可以放在不同房间,可以用钥匙开门,也可以选择破门而入取钥匙,但第一间房子不可以被破门而入。最多破门 k 次,求能进入所有房间的概率。

第一类stirling数。

[ s1(n,i)- s1(n - 1,i - 1)] (i = 1 ~ k)求和,再除 n!即可。

#include
#include
#include
#include
#define ll long long
using namespace std;
const int maxn = 20 + 5;
ll s1[maxn][maxn];
//  calculate factorial
ll fact(int n)
{
    ll ans = 1;
    while(n)
    {
        ans *= n;
        --n;
    }
    return ans;
}
void init_s1()
{
    memset(s1,0,sizeof(s1));
    for(int i = 1;i <= 20; ++i) s1[i][i] = 1;
    for(int i = 2;i <= 20; ++i)
    {
        for(int j = 1;j < i; ++j)
        {
            s1[i][j] = (s1[i - 1][j - 1] + (i - 1) * s1[i - 1][j]);
        }
    }
}
int main()
{
    init_s1();
    int t,n,k;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d",&n,&k);
        ll t1 = 0,t2 = 0;
        for(int i = 1;i <= k; ++i) t1 += (s1[n][i] - s1[n - 1][i - 1]);
        t2 = fact(n);
        double ans = (double)t1 / t2;
        printf("%.4f\n",ans);
    }
    return 0;
}

你可能感兴趣的:(数学相关)