基础dp

问题 A: 谷神的赌博游戏

时间限制: 1 Sec   内存限制: 16 MB
提交: 414   解决: 112
[ 提交][ 状态][ 讨论版]

题目描述

  NEUQ的谷神要和我赌一个游戏：谷神要求我随机在纸上写出整数集合{1,2,3，...,3n+1}（n是整数）的一个排列（即不重复的随机写出从1到3n+1的所有整数）。并且要求在我写的过程中，从我写的第一个数开始一直加到我正在写的数的总和不被3整除。如果我能写出来符合要求的一个排列，那么我就赢得游戏。那么问题来了，我赢得游戏的概率是多少？

输入

一组测试数据，第一行输入测试样例的数目k，接下来k行每行一个正整数n代表一个样例（1<=n<=15）。

输出

对于每个样例数据依次输出我赢得比赛的概率（结果保留小数点后9位有效数字）。

样例输入

1
1

样例输出

0.250000000

提示

例如n=1，则谷神要求我随机写1到4的排列，如果我按顺序写1 3 4 2则是合法的，因为1，1+3、1+3+4、1+3+4+2都不被3整除。如果我按顺序写1 2 3 4则是不合法的，因为当我写到2的时候1+2=3可以被3整除，不符合游戏规定。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 20;
double dp[maxn][maxn][maxn];
int main()
{
    int Tcase;
    scanf("%d",&Tcase);
    while(Tcase --)
    {
        int n;
        scanf("%d",&n);
        memset(dp,0,sizeof(dp));
        dp[0][0][0] = 1;
//        cout << dp[1][0][0] << " " << dp[0][1][0] << endl;
        for(int i = 0; i <= n + 1; i ++)
        {
            for(int j = 0; j <= n; j ++)
            {
                if( (i + j * 2) % 3 == 0)
                continue;
                for(int k = 0; k <= n; k ++)
                {

                    dp[i][j][k] = dp[i - 1][j][k] * (n - i  + 2) * 1.0 / (3 * n + 2 - i - j - k) +
                                   dp[i][j - 1][k] * (n - j + 1) * 1.0 / (3 * n + 2 - i - j - k) +
                                   dp[i][j][k - 1] * (n - k + 1) * 1.0 / (3 * n + 2 - i - j - k);
//                    cout << i << " " << j << " " << k << " " << dp[i][j][k] << endl;
                }

            }
        }
        printf("%.9f\n",dp[n + 1][n][n]);
    }
    return 0;
}