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;
    }
};
相关推荐
你撅嘴真丑2 小时前
第九章-数字三角形
算法
uesowys2 小时前
Apache Spark算法开发指导-One-vs-Rest classifier
人工智能·算法·spark
ValhallaCoder2 小时前
hot100-二叉树I
数据结构·python·算法·二叉树
董董灿是个攻城狮2 小时前
AI 视觉连载1:像素
算法
智驱力人工智能3 小时前
小区高空抛物AI实时预警方案 筑牢社区头顶安全的实践 高空抛物检测 高空抛物监控安装教程 高空抛物误报率优化方案 高空抛物监控案例分享
人工智能·深度学习·opencv·算法·安全·yolo·边缘计算
孞㐑¥3 小时前
算法——BFS
开发语言·c++·经验分享·笔记·算法
月挽清风3 小时前
代码随想录第十五天
数据结构·算法·leetcode
XX風4 小时前
8.1 PFH&&FPFH
图像处理·算法
NEXT064 小时前
前端算法:从 O(n²) 到 O(n),列表转树的极致优化
前端·数据结构·算法
代码游侠4 小时前
学习笔记——设备树基础
linux·运维·开发语言·单片机·算法