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;
    }
};
相关推荐
你们补药再卷啦14 分钟前
人工智能算法概览
人工智能·算法
cnxy18819 分钟前
围棋对弈Python程序开发完整指南:步骤3 - 气(Liberties)的计算算法设计
python·算法·深度优先
AndrewHZ29 分钟前
【图像处理基石】什么是光栅化?
图像处理·人工智能·算法·计算机视觉·3d·图形渲染·光栅化
小白菜又菜37 分钟前
Leetcode 944. Delete Columns to Make Sorted
算法·leetcode
我找到地球的支点啦1 小时前
Matlab系列(006) 一利用matlab保存txt文件和读取txt文件
开发语言·算法·matlab
Dev7z1 小时前
基于Matlab实现GRACE卫星重力数据的全球水储量变化估算与分析
人工智能·算法·matlab
爱喝热水的呀哈喽2 小时前
11题目汇总
算法
三斗米2 小时前
Transformer入门:一文读懂《Attention Is All You Need》
算法·架构
Swift社区2 小时前
LeetCode 458 - 可怜的小猪
算法·leetcode·职场和发展
AI科技星3 小时前
宇宙的像素:真空中一点如何编码无限星光
数据结构·人工智能·算法·机器学习·重构