数字和方法数

题目

给定一个有$n$个正整数的数组$array$和一个整数$sum$，求选择数组$array$中部分数字和为$sum$的方案数。当两种选取方案有一个数字的下标不一样，我们就认为是不同的组成方案。 输入为两行，第一行为两个正整数$n$(1 ≤ $n$ ≤ 1000)，$sum$(1 ≤ $sum$ ≤ 1000)，第二行为$n$个正整数(32位整数)，以空格隔开。

• 测试输入
  5 15
  5 5 10 2 3
• 测试输出
  4

分析1

如果针对每个数字是否选择，可以使用简单的回溯法，如果当前和等于$sum$则方法数递增。如果当前和已经超过了$sum$则回溯，否则继续尝试下一个数字。由于方法数可能很多，所以应该使用long型存储。

代码1

import java.util.Scanner;

public class Summary {
    static long method = 0;

    static void solution(int acc, int i, int sum, int[] array) {
        if (acc == sum) {
            ++method;
        } else if (acc < sum) {
            if (i < array.length) {
                solution(acc + array[i], i + 1, sum, array);// 选择第i个
                solution(acc, i + 1, sum, array);// 不选择第i个
            }
        }
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int sum = sc.nextInt();
        int[] array = new int[n];
        for (int i = 0; i < n; i++) {
            array[i] = sc.nextInt();
        }
        solution(0, 0, sum, array);
        System.out.println(method);
    }
}

分析2

如果从后向前考虑，为了得到一个较大的$sum$的最多方案数，需要保证之前较小的$sum$的已经得到最多方案数，从而具有最优子结构。这类似于背包问题，只是背包要求装满，且求的是装进背包的方案数。定义$m[i][j]$表示从前$i$个数中选出部分数字和为$j$的方案数，对于第$i$个数可选可不选，方案数为两种情况之和。基本情况的初始化比较关键，当$i=0$时，除了$j=0$的方案数为1，其他情况均不可能实现，所以方案数为0。当$j=0$时，由于$array$都是正整数，所以只能不选择任何数字，所以方案数为1。递归式为

$$m[i][j]=\left\{\begin{array}{ll} 0, & i=0,j>0\\ 1, & j=0\\ m[i-1][j]+m[i-1][j-array[i]], & i,j>0 \end{array}\right.$$

代码2

import java.util.Scanner;

public class Summary {
    static void solution(int[] array, long[][] m, int sum) {
        int n = array.length;
        for (int i = 1; i <= sum; i++) {
            m[0][i] = 0;
        }
        for (int i = 0; i <= n; i++) {
            m[i][0] = 1;
        }
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= sum; j++) {
                if (j - array[i - 1] >= 0) {
                    m[i][j] = m[i - 1][j] + m[i - 1][j - array[i - 1]];
                } else {
                    m[i][j] = m[i - 1][j];
                }
            }
        }
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int sum = sc.nextInt();
        int[] array = new int[n];
        for (int i = 0; i < n; i++) {
            array[i] = sc.nextInt();
        }
        long[][] m = new long[n + 1][sum + 1];
        solution(array, m, sum);
        System.out.println(m[n][sum]);
    }
}