leetcode算法题--掷骰子的N种方法

原题链接:https://leetcode-cn.com/problems/number-of-dice-rolls-with-target-sum/

动态规划

dp[i][k]表示前i个骰子,k表示目标数字

状态转移

//前i个骰子掷数目k的种类数等于前i-1个骰子掷数目k-j的种类数,j为第i个骰子掷的数字
dp[i][k]+=dp[i-1][k-j]

代码:

int numRollsToTarget(int d, int f, int target) {
     
    const int mod=1000000007;
    vector<vector<int>> dp(d+1,vector<int>(max(f+1,target+1),0));
    for(int k=1;k<=min(f,target);k++){
     
        dp[1][k]=1;//第一个骰子可以掷的数目
    }
    for(int i=2;i<=d;i++){
     
        for(int k=1;k<=target;k++){
     //目标数字
            for(int j=1;j<=min(k,f);j++){
     //第i个骰子掷的数字
                dp[i][k]=(dp[i][k]+dp[i-1][k-j])%mod;
            }
        }
    }
    return dp[d][target];
}

你可能感兴趣的:(算法)