Leetcode 377 - Combination Sum IV(DAG上的dp)

来源:互联网 发布:北大青鸟java班学费 编辑:程序博客网 时间:2024/06/05 23:04

题意

给定一个数n和一个数组nums,要求用nums内的数去组成n,问多少种解法。顺序不同也算新的解法。

思路

其实就一DAG上路径统计问题
我们可以这样考虑:把0到target上的每个值看做一个点,如果能够从点A到点B的话, 我们就相当于可以连一条边,从点A到点B能够连边的充要条件是nums内存在一个数是B-A的值。然后就成了DAG上路径统计问题了。

我们这样定义状态:d[j],当前需要凑出来的钱是j的总方案数。

转移方程d[j]=k=0n1d[jak]jak

Follow up

如果nums内包含负数该怎么办,如果有负数我们要给这道题添加什么限制?

如果我们nums内有负数存在的话,就意味着我们的结果会有无穷多个,我们可以添加限制每个数的使用次数。

代码

const int maxn = 100000 + 5;int d[maxn];class Solution {public:    int combinationSum4(vector<int>& nums, int target) {        memset(d, 0, sizeof(d));        d[0] = 1; int n = nums.size();        for (int j = 1; j <= target; j++) {            for (int k = 0; k < n; k++) {                if (j >= nums[k]) {                    d[j] += d[j - nums[k]];                }            }        }        return d[target];    }};
0 0
原创粉丝点击