1155掷骰子等于目标和的方法数 (dfs + 记忆化搜索)

cpp 复制代码
class Solution {
public:
    int ans = 0;
    int Mod = 1e9 + 7;

    int numRollsToTarget(int n, int k, int target) {
        return dfs(n, k, target);
    }

    map<string, int> m;
    int dfs(int n, int k, int target) {
        if(target == 0 && n == 0) return 1;
        if(target < 0 || n == 0) return 0;

        // 遍历第index个骰子的第j个面
        // 这一步是记忆化搜索的过程,剩下n个骰子,剩下的target肯定有重复搜索的情况
        // 比如,剩下一个骰子,前面的结果是 2 3 和 3 2 就是导致 1 1 被重复搜索
        // 所以,记录下现在的 n k target 三元组,避免后面再次遇见的时候,重复搜索 
        string s = to_string(n) + "_" + to_string(k) + "_" + to_string(target);
        // cout << s << endl;

        if(m.count(s)) {
            // cout << s << endl;
            return m[s];
        }

        int res = 0;
        for(int i = 1; i <= k; i ++) {
            int t = dfs(n - 1, k, target - i);
            res = (res + t) % Mod;
        }
        m[s] = res;
        return res;
    }
};
相关推荐
tobias.b1 分钟前
408真题解析-2010-9-数据结构-折半查找的比较次数
java·数据结构·算法·计算机考研·408真题解析
源代码•宸2 分钟前
Leetcode—404. 左叶子之和【简单】
经验分享·后端·算法·leetcode·职场和发展·golang·dfs
WBluuue7 分钟前
数据结构与算法:dp优化——优化尝试和状态设计
c++·算法·leetcode·动态规划
im_AMBER26 分钟前
Leetcode 105 K 个一组翻转链表
数据结构·学习·算法·leetcode·链表
sin_hielo29 分钟前
leetcode 1877
数据结构·算法·leetcode
睡不醒的kun1 小时前
定长滑动窗口-基础篇(2)
数据结构·c++·算法·leetcode·职场和发展·滑动窗口·定长滑动窗口
庄小焱1 小时前
【机器学习】——房屋销售价格预测实战
人工智能·算法·机器学习·预测模型
txzrxz1 小时前
单调栈详解(含题目)
数据结构·c++·算法·前缀和·单调栈
AI科技星1 小时前
张祥前统一场论的数学表述与概念梳理:从几何公设到统一场方程
人工智能·线性代数·算法·机器学习·矩阵·数据挖掘
程序员-King.1 小时前
day167—递归—二叉树的直径(LeetCode-543)
算法·leetcode·深度优先·递归