【力扣刷题】1155. 掷骰子等于目标和的方法数-dp
这里有 n 个一样的骰子,每个骰子上都有 k 个面,分别标号为 1 到 k 。
给定三个整数 n , k 和 target ,返回可能的方式(从总共 kn 种方式中)滚动骰子的数量,使正面朝上的数字之和等于 target 。
答案可能很大,你需要对 109 + 7 取模 。
示例 1:
输入:n = 1, k = 6, target = 3
输出:1
解释:你扔一个有 6 个面的骰子。
得到 3 的和只有一种方法。
1
2
3
4
示例 2:
输入:n = 2, k = 6, target = 7
输出:6
解释:你扔两个骰子,每个骰子有 6 个面。
得到 7 的和有 6 种方法:1+6 2+5 3+4 4+3 5+2 6+1。
1
2
3
4
示例 3:
输入:n = 30, k = 30, target = 500
输出:222616187
解释:返回的结果必须是对 109 + 7 取模。
1
2
3
提示:
1 <= n, k <= 30
1 <= target <= 1000
class Solution {
public:
int numRollsToTarget(int n, int k, int target) {
vector<vector<int>>dp(n+1,vector<int>(target+1));
const int MOD=1e9+7;
dp[0][0]=1;
for(int i=1;i<=n;i++){
for(int j=1;j<=k;j++){
for(int x=0;x<=target;x++){
if(x-j>=0){
dp[i][x]=(dp[i][x]+dp[i-1][x-j])%MOD;
}
}
}
}
return dp[n][target];
}
};
/*
今日总结:
1. 本题抄的官方题解,而且还看不懂后面的优化……
2. 开vector时要定上大小比较好
*/
版权声明:
作者:Zhang, Hongxing
链接:http://zhx.info/archives/153
来源:张鸿兴的学习历程
文章版权归作者所有,未经允许请勿转载。