kSum

25 篇文章 0 订阅

参考点击打开链接

原文说的很好,尤其是

使用三维动规数组dp[i][j][t],表示从0遍历到A[i]后找到的j个元素之和为t的情况的总数。最后返回从整个A数组找到的k个元素之和为target的情况总数即可

其余的就过遍数即可。

/**

     * @param A: an integer array.
     * @param k: a positive integer (k <= length(A))
     * @param target: a integer
     * @return an integer
     */
    public int kSum(int A[], int k, int target) {
        // write your code here
        if (A == null || A.length < k) {
            return 0;
        }
        int [][][] matrix = new int[A.length + 1][k + 1][target + 1];
        for (int i = 0; i <= A.length; i++) {
            matrix[i][0][0] = 1;
        }
        for (int i = 1; i <= A.length; i++) {
            for (int j = 1; j <= k && j <= i; j++) {
                for (int t = 1; t <= target; t++) {
                    matrix[i][j][t] = matrix[i - 1][j][t];
                    if (A[i - 1] <= t) {
                        matrix[i][j][t] = matrix[i][j][t] + matrix[i - 1][j - 1][t - A[i - 1]];
                    }
                }
            }
        }
        return matrix[A.length][k][target];
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值