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;
    }
};
相关推荐
君义_noip3 小时前
信息学奥赛一本通 1661:有趣的数列 | 洛谷 P3200 [HNOI2009] 有趣的数列
c++·算法·组合数学·信息学奥赛·csp-s
程序员:钧念3 小时前
深度学习与强化学习的区别
人工智能·python·深度学习·算法·transformer·rag
英英_4 小时前
MATLAB数值计算基础教程
数据结构·算法·matlab
一起养小猫4 小时前
LeetCode100天Day14-轮转数组与买卖股票最佳时机
算法·leetcode·职场和发展
hele_two4 小时前
快速幂算法
c++·python·算法
l1t5 小时前
利用DeepSeek将python DLX求解数独程序格式化并改成3.x版本
开发语言·python·算法·数独
jllllyuz5 小时前
基于子集模拟的系统与静态可靠性分析及Matlab优化算法实现
算法·matlab·概率论
程序员-King.5 小时前
day143—递归—对称二叉树(LeetCode-101)
数据结构·算法·leetcode·二叉树·递归
BlockChain8885 小时前
字符串最后一个单词的长度
算法·go
爱吃泡芙的小白白5 小时前
深入解析:2024年AI大模型核心算法与应用全景
人工智能·算法·大模型算法