[leetcode] Combination Sum IV

cocolalala

于 2016-07-26 01:34:19 发布

阅读量447

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： c++ leetcode 算法搜索

本文链接：https://blog.csdn.net/yezizp/article/details/52031837

算法专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种使用记忆化搜索方法解决组合问题的算法实现。通过对输入数组进行排序并利用辅助数组记录中间结果的方式，有效地减少了重复计算，提高了求解以特定值开始的组合数目问题的效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：记忆化搜索，定义函数查找以index开始，值为target的个数，顺便对nums排序减少不必要的搜索。

代码如下：

class Solution {
public:
    int combinationSum4(vector<int>& nums, int target) {
        int result = 0;
        if(nums.empty() || target == 0) return result;
        sort(nums.begin(), nums.end());
        vector<vector<int> > path(nums.size(), vector<int>(target + 1, -1));
        for(int i = 0; i < nums.size(); ++i)
            result += get(i, target, path, nums);
        return result;
    }
private:
    int get(int index, int target, vector<vector<int> > &path, vector<int>& nums) {
        if(target == nums[index]) path[index][target] = 1;
        if(target < nums[index]) path[index][target] = 0;
        if(path[index][target] != -1)
            return path[index][target];
        int cnt = 0;
        target -= nums[index];
        for(int i = 0; i < nums.size(); ++i)
            cnt += get(i, target, path, nums);
        path[index][target + nums[index]] = cnt;
        return cnt;
    }
};