【LeetCode每日一题】377. 组合总和 Ⅳ

题目

给你一个由 不同 整数组成的数组 nums ，和一个目标整数 target 。请你从 nums 中找出并返回总和为 target 的元素组合的个数。

题目数据保证答案符合 32 位整数范围。

示例1

输入：nums = [1,2,3], target = 4
输出：7
解释：
所有可能的组合为：
(1, 1, 1, 1)
(1, 1, 2)
(1, 2, 1)
(1, 3)
(2, 1, 1)
(2, 2)
(3, 1)
请注意，顺序不同的序列被视作不同的组合。

题解

1.记忆化搜索

递归并采用数组存储每个target的可能性有多少种。

class Solution {
public:
    vector<int> memo;
    int combinationSum4(vector<int>& nums, int target) {
        int n = nums.size();
        memo = vector<int>(target + 1, -1);
        return dfs(nums, target);
    }

    int dfs(vector<int>& nums, int target) {
        if (target == 0) return memo[target] = 1;
        if (memo[target] != -1) return memo[target];
        int res = 0;
        
        for (auto num : nums) {
            if (target - num >= 0) 
                res += dfs(nums, target - num);
        }

        return memo[target] = res;
    }
};

2.动态规划

以dp[i]表示当target=i时，所有的可能性。

那么转移方程为：

dp[i] += dp[i-num];

实现如下：

class Solution {
public:
    int combinationSum4(vector<int>& nums, int target) {
        vector<unsigned long long> dp(target + 1, 0);
        dp[0] = 1;
        for (int i = 1; i <= target; i++) {
            for (auto num : nums) {
                if (i - num >= 0) {
                    dp[i] += dp[i - num];
                }
            }
        }
        return dp[target];
    }
};