lintcode k数和

作者: yzawyx0220 | 来源:发表于2017-01-18 16:04 被阅读188次

给定n个不同的正整数,整数k(k < = n)以及一个目标数字。    
在这n个数里面找出K个数,使得这K个数的和等于目标数字,求问有多少种方案?
样例
给出[1,2,3,4],k=2, target=5,[1,4] and [2,3]是2个符合要求的方案
题目链接:http://www.lintcode.com/zh-cn/problem/k-sum/

这类问题一般都是使用动态规划的方法。建立到一个三维数组dp[i][j][k]表示在前i个数里取j个数和为k的方案数,于是dp[i][j][k] = dp[i-1][j][k] + dp[i-1][j-1][k-A[i]],即包括A[i]和不包括A[i]的两种方案。可以看到i只跟i-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<int> > dp(k + 1,vector<int>(target + 1,0));
        dp[0][0] = 1;
        for (auto num : A) {
            for (int i = k;i >= 1;i--) {
                for (int j = target;j >= num;j--) {
                    dp[i][j] += dp[i - 1][j - num];
                }
            }
        }
        return dp[k][target];
    }
};

相关文章

  • lintcode k数和

    给定n个不同的正整数,整数k(k < = n)以及一个目标数字。在这n个数里面找出K个数,使得这K个数的和等于目标...

  • lintcode k数和||

    给定n个不同的正整数,整数k(1<= k <= n)以及一个目标数字。在这n个数里面找出K个数,使得这K个数的和等...

  • lintcode-k数和

    动态规划(确定0-1背包、完全背包、多重背包)0-1背包:每个元素要么出现,要么不出现,逆序遍历,数组定义为:前i...

  • 90 lintcode k数和

    class Solution { public: /* * @param A: an integer a...

  • Remove K Digits

    https://www.lintcode.com/problem/remove-k-digits/description

  • k数和

    def kSum(self, A, k, target):n = len(A)if n <= 0 or k <= ...

  • K数和

    题目: 给定n个不同的正整数,整数k(k < = n)以及一个目标数字。 在这n个数里面找出K个数,使得这K个数的...

  • LintCode 5. Kth Largest Element

    原题 LintCode 5. Kth Largest Element Description Find K-th ...

  • 第k大元素

    (lintcode上面的题解)第k大元素:(从小到大的排序)

  • lintcode 丑数

    设计一个算法,找出只含素因子2,3,5 的第 n 大的数。直接寻找丑数,由定义可知,丑数是由2m,3n,5^l,因...

网友评论

    本文标题:lintcode k数和

    本文链接:https://www.haomeiwen.com/subject/wkcbbttx.html