Jianzhiはインタビューの質問60を提供します。n個のサイコロの数[単純]-動的計画法

インタビューの質問60。n個のサイコロの数

n個のサイコロに対してn〜6n個の結果があります

各結果は、n-1個のサイコロの結果に関連しています

最初の発生をカウントしてから、確率を計算します

class Solution {
public:
    vector<double> twoSum(int n) {
        vector<vector<int>> dp(n+1,vector<int>(6*n+1,0));
        for(int i=1;i<=6;i++)
            dp[1][i]=1;
        for(int i=2;i<=n;i++){
            for(int j=i;j<=6*i;j++){
                for(int k=1;k<=6;k++){
                    if(j-k<=0)   break;
                    dp[i][j]+=dp[i-1][j-k];
                }
            }
        }
        vector<double> res;
        int tmp=pow(6,n);
        for(int i=n;i<=6*n;i++){
            res.push_back(dp[n][i]*1.0/tmp);
        }
        return res;
    }
};