Leetcode 1155. 掷骰子的N种方法(149周赛)

这里有 d 个一样的骰子,每个骰子上都有 f 个面,分别标号为 1, 2, ..., f。

我们约定:掷骰子的得到总点数为各骰子面朝上的数字的总和。

如果需要掷出的总点数为 target,请你计算出有多少种不同的组合情况(所有的组合情况总共有 f^d 种),模 10^9 + 7 后返回。

示例 1:

输入:d = 1, f = 6, target = 3
输出:1
示例 2:

输入:d = 2, f = 6, target = 7
输出:6
示例 3:

输入:d = 2, f = 5, target = 10
输出:1
示例 4:

输入:d = 1, f = 2, target = 3
输出:0
示例 5:

输入:d = 30, f = 30, target = 500
输出:222616187
 

提示:

1 <= d, f <= 30
1 <= target <= 1000

    这道题典型的动态规划,通过建立DP数组Val记录每一种状态存在的可能性,其中纵坐标代表筛子数量,横坐标代表目标值,值代表可能性的数量,其中每一个都可以转换为筛子数-1,目标值减去一个筛子的可能值(1到f)的可能性之和。

class Solution {
public:
	int Val[31][1001];
	int mod = 1e9 + 7;

	int numRollsToTarget(int d, int f, int target) {
		memset(Val, 0, sizeof(Val));

		for (int i = 1; i <= f; i++)
		{
			Val[1][i] = 1;
		}
		for (int i = 2; i <= d; i++)
		{
            for(int k = 1;k<=target;k++)
            {
                 for (int j = 1; j <= f && j < k; j++)
                {
                    Val[i][k] = (Val[i][k] + Val[i - 1][k - j]) % mod;
                }   
            }
		}
		return Val[d][target];
	}
};
发布了104 篇原创文章 · 获赞 3 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/Yanpr919/article/details/99196728