LintCode-k数和

LintCode-k数和

给定n个不同的正整数,整数k(k < = n)以及一个目标数字。 

在这n个数里面找出K个数,使得这K个数的和等于目标数字,求问有多少种方案?

样例
给出[1,2,3,4],k=2, target=5,[1,4] and [2,3]是2个符合要求的方案

用dp解,d[i][j]内含一个map,表示前j+1个数中取i+1个组合成各个和(将大于n的去掉)的组合数量
i=j=0时,d[i][j]={{A[j],1}}
i=0,j≠0时,d[i][j]=d[i][j-1]∪{A[j],1};
i≠0时,d[i][j]=(d[i-1][j-1]+A[j])∪d[i][j-1];

class Solution {
public:
    /**
    * @param A: an integer array.
    * @param k: a positive integer (k <= length(A))
    * @param target: a integer
    * @return an integer
    */
    int kSum(vector<int> A, int k, int target) {
        // wirte your code here
        vector<vector<map<int, int>>> dp(k);
        for (int i = 0; i<k; i++) {
            dp[i].resize(A.size());
        }
        for (int i = 0; i<k; i++) {
            for (int j = i; j<A.size(); j++) {
                if (i == 0) {
                    if (j == 0) {
                        dp[i][j][A[j]]++;
                    }
                    else {
                        dp[i][j] = dp[i][j - 1];
                        dp[i][j][A[j]]++;
                    }
                }
                else {
                    for (auto m : dp[i - 1][j - 1]) {
                        if (m.first + A[j] <= target)
                            dp[i][j][m.first + A[j]] += m.second;
                    }
                    for (auto m : dp[i][j - 1]) {
                        dp[i][j][m.first] += m.second;
                    }
                }
            }
        }
        return dp[k - 1][A.size() - 1][target];
    }
};

一开始用三维数组溢出了,后来改成map,由于d[i][j]只和d[i-1]有关,其实还可以在空间上进行优化。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值