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;
    }
};
相关推荐
徐子童8 分钟前
数据结构----排序算法
java·数据结构·算法·排序算法·面试题
hansang_IR15 分钟前
【记录】四道双指针
c++·算法·贪心·双指针
_OP_CHEN18 分钟前
算法基础篇:(十二)基础算法之倍增思想:从快速幂到大数据运算优化
大数据·c++·算法·acm·算法竞赛·倍增思想
CoovallyAIHub39 分钟前
分割万事万物的AI,再进化!Meta SAM 3 来了,支持中文提示词!
深度学习·算法·计算机视觉
九年义务漏网鲨鱼39 分钟前
蓝桥杯算法——记忆化搜索
算法·职场和发展·蓝桥杯
04aaaze1 小时前
C++(C转C++)
c语言·c++·算法
Swift社区1 小时前
LeetCode 429 - N 叉树的层序遍历
算法·leetcode·职场和发展
星释1 小时前
Rust 练习册 32:二分查找与算法实现艺术
开发语言·算法·rust
zl_vslam1 小时前
SLAM中的非线性优-3D图优化之四元数在Opencv-PNP中的应用(五)
人工智能·算法·计算机视觉
机器学习之心1 小时前
经典粒子群优化算法PSO-LSTM回归+SHAP分析+多输出+新数据预测!Matlab代码实现
算法·lstm·pso-lstm·shap分析