两数之和&三数之和&四数之和&K数之和

2020-11-15  本文已影响0人  Skittle丶

今天看了一道谷歌K数之和的算法题,忽然想起来之前在力扣上做过2、3、4数之和的题,觉得很有必要来整理一下。
其实2、3、4数之和是一类题,用多个指针即可完成;K数之和是一类题,又是我学的最不好的DP。
假设数组为A,要找的分别是a、b、c、d,目标为target
2数之和:固定aPoint。bPoint去找target-A[aPoint];
3数之和:固定aPoint。bPoint和cPoint组成一个二数之和的问题,target=target-A[aPoint];
4数之和:固定aPoint,bPoint。cPoint,dPoint组成一个二数之和的问题。
K数之和:设状态为f[i][j][p],表示前i个数中找出j个数且和等于p的方案数目。
这个问题的关键在于到底要不要当前这个数A[i]。
也就是说,若要满足当前目标currentTarget,一共有两条路:
currentTarget的方案数=不要A[i]的方案数+要A[i]的方案数(当然A[i]要比当前目标值小,才能要)

public class Solution {
    public int kSum(int[] A, int k, int target) {
        if(A == null || A.length ==0 || k <= 0)
            return 0;
        //正整数的总数从0 - A.length; 可选的个数从 0 - k; target也是从 0 - target
        int[][][] f = new int[A.length + 1][ k + 1][target + 1];
        //初始值,
        for(int i = 0; i <= A.length; i ++){
            f[i][0][0] = 1;
        }
        //f[i][j][t] = f[i - 1][j][t] + f[i - 1][j - 1][t -A[i]]
        for(int i = 1; i <= A.length; i ++)
            for(int j = 1; j <= k; j ++)
                for(int t = 1; t <= target; t ++){
                    //先更新当前的这个选项,还没有把第i个数选择进来
                    f[i][j][t] = f[i - 1][j][t];
                    //如果想把第i个数选进来,就要看第i个数是否大于目前的目标值
                    f[i][j][t] += t - A[i - 1] >=0 ? f[i - 1][j - 1][t -A[i - 1]] : 0;
                }
        return f[A.length][k][target];
    }
}
上一篇 下一篇

猜你喜欢

热点阅读