1155. 掷骰子等于目标和的方法数
力扣每日1题。
思路
第一反应就是当投一个骰子,骰子数会减1,和会减骰出数,分析数量关系,可拆分为子问题(DFS+DP),考虑统计子问题会骰出的方法数(递归?)。
DFS
该方法会超时,时间复杂度为O(k^n),n为骰子数量,k是骰子面数。
class Solution {
final int MOD = (int) 1e9 + 7;
public int numRollsToTarget(int n, int k, int target) {
// 由数量关系知可分成子问题,DFS或DP
// DFS
int[][] res = new int[n + 1][target + 1];// 使用二维数组res来存储统计的满足方案
return DFS(n, k, target, res);
}
public int DFS(int n, int k, int target, int[][] res) {
if (target < 0) return 0;
if (n == 0) return target == 0 ? 1 : 0;
if (res[n][target] != 0) {
return res[n][target];
}
int count = 0;
for (int i = 1; i <= k; i++) {
// 遍历面统计满足次数
count = (count + DFS(n - 1, k, target - i, res)) % MOD;
}
res[n][target] = count;
return count;
}
}
DP
优化DFS代码,用DP表(res二维数组)模拟递归。
注意子问题拆分后的边界条件!!!
class Solution {
final int MOD = (int) 1e9 + 7;
public int numRollsToTarget(int n, int k, int target) {
// DP
int[][] res = new int[n + 1][target + 1];
// 只有1个骰子,骰子投出目标点数只有一种方案(注意和可能比面数小)
for (int i = 1; i <= k && i <= target; i++) {
res[1][i] = 1;
}
// 从2个骰子开始,统计子问题的满足方案
for (int i = 2; i <= n; i++) {
// 拆分子问题
for (int j = 1; j <= target; j++) {
// 遍历当前骰子的每个面(注意组合骰子数会出现和大于面数)
for (int m = 1; m <= k && m <= j; m++) {
// 当前骰子的子问题满足方案是上一个骰子骰出的减去当前骰子的骰出数的和。
res[i][j] = (res[i][j] + res[i - 1][j - m]) % MOD;
}
}
}
return res[n][target];
}
}