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;
    }
};
相关推荐
程芯带你刷C语言简单算法题几秒前
Day33~实现一个算法来识别一个字符串。
c语言·算法·c
学编程就要猛几秒前
算法:2.复写零
java·数据结构·算法
TL滕1 分钟前
从0开始学算法——第二十一天(链表练习)
笔记·学习·算法
LYFlied2 分钟前
【每日算法】LeetCode238. 除自身以外数组的乘积
数据结构·算法·leetcode·面试·职场和发展
仰泳的熊猫3 分钟前
1154 Vertex Coloring
数据结构·c++·算法·pat考试
_OP_CHEN3 分钟前
【算法基础篇】(三十七)图论基础之单源最短路:从原理到实战,4 大算法彻底吃透!
算法·图论
a程序小傲5 分钟前
京东Java面试被问:垃圾收集算法(标记-清除、复制、标记-整理)的比较
java·算法·面试
元亓亓亓12 分钟前
LeetCode热题100--118. 杨辉三角--简单
算法·leetcode·职场和发展
耶叶13 分钟前
查找算法学习总结2:代码分析篇
数据结构·学习·算法
StudyWinter27 分钟前
【c++】thread总结
开发语言·c++·算法