动态规划之神奇的口袋

题目描述

有一个神奇的口袋，总的容积是40，用这个口袋可以变出一些物品，这些物品的总体积必须是40。John现在有n个想要得到的物品，每个物品的体积分别是a1，a2……an。John可以从这些物品中选择一些，如果选出的物体的总体积是40，那么利用这个神奇的口袋，John就可以得到这些物品。现在的问题是，John有多少种不同的选择物品的方式。

输入描述:

输入的第一行是正整数n (1 <= n <= 20)，表示不同的物品的数目。接下来的n行，每行有一个1到40之间的正整数，分别给出a1，a2……an的值。

输出描述:

输出不同的选择物品的方式的数目。

示例1

输入

复制

3
20
20
20

输出

复制

3

代码如下：（运用动态规划的思想来解决问题）

package www.light.long1;

import java.util.Scanner;

/**
 * Created with IntelliJ IDEA
 * Description:
 * User:Light_Long
 * Date:2019-07-05
 * Time:19:44
 */
public class Main {
        public static void main(String[] args) {
            Scanner sc = new Scanner(System.in);
            while(sc.hasNextInt()){
                int n = sc.nextInt();
                int[] arr = new int[n+1];
                //该数组第一个元素为0，其余元素为要输入的元素
                arr[0] = 0;
                for(int i = 1;i<=n;i++){
                    arr[i] = sc.nextInt();
                }
                System.out.println(nNumAddSumCount(arr,40));
            }

        }

        public static long nNumAddSumCount(int[] arr,int sum){
            int n = arr.length;
            //定义二维数组，用来表示方法数
            long[][] dp = new long[n+1][sum+1];
            //对第一列全部赋1
            for(int i = 0;i<=n;i++){
                dp[i][0] = 1;
            }
            //对第一行除第一个位置外其余赋0
            for(int i = 1;i<=sum;i++){
                dp[0][i] = 0;
            }
            for(int i = 1;i<=n;i++){
                for(int j = 1;j<=sum;j++){
                    //如果arr[i]>j，此时(i,j)位置的值为(i-1,j)位置的值
                    if(arr[i-1]>j){
                        dp[i][j] = dp[i-1][j];
                    }else{
                        dp[i][j] = dp[i-1][j]+dp[i-1][j-arr[i-1]];
                    }
                }
            }
            return dp[n][sum];
        }
}