蓝桥杯2021 砝码称重 动态规划 (Java)

你有一架天平和 NN 个砝码，这 NN 个砝码重量依次是 W_1, W_2, · · · , W_NW1​,W2​,⋅⋅⋅,WN​。

请你计算一共可以称出多少种不同的重量？ 注意砝码可以放在天平两边。

输入格式

输入的第一行包含一个整数 NN。

第二行包含 NN 个整数：W_1, W_2, W_3, · · · , W_NW1​,W2​,W3​,⋅⋅⋅,WN​。

输出格式

输出一个整数代表答案。

样例输入

3
1 4 6

样例输出

10

样例说明

能称出的 1010 种重量是：1、2、3、4、5、6、7、9、10、111、2、3、4、5、6、7、9、10、11​。

1 = 1；1=1；

2 = 6 − 4 (2=6−4(天平一边放 66，另一边放 4)；4)；​

3 = 4 − 1；3=4−1；

4 = 4；4=4；

5 = 6 − 1；5=6−1；​

6 = 6；6=6；

7 = 1 + 6；7=1+6；

9 = 4 + 6 − 1；9=4+6−1；

10 = 4 + 6；10=4+6；

11 = 1 + 4 + 6。11=1+4+6。

评测用例规模与约定

对于 50%50的评测用例，1 ≤ N ≤ 151≤N≤15。

对于所有评测用例，1 ≤ N ≤ 100,N1≤N≤100,N​个砝码总重不超过 100000100000。

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

思路：

砝码的放置可以是左右放置，假设物品都是放在做盘，然后砝码放在右盘和左盘都是可以的。在这设定为，砝码放在右盘是相加，则砝码放到左盘的时候，物品的质量就需要右盘砝码的质量减去左盘砝码的质量。起初假定所有的砝码都放在右盘，物品都放在左盘，这是一个01背包问题，求出砝码只能放置到右盘的能凑出来的数。然后计算当砝码可以放到左盘的时候，起初砝码全部放到右盘，能够凑出来的数字，是由砝码重量几个数字凑出来的，在能凑出来的数字情况下减掉其中的砝码的重量，就是相当于砝码可以放置到左盘的情况，就是砝码放置到左盘的时候能够凑出来的数。

开辟二维dp数组的时候要注意，将能凑出来的数字放大很多，因为在计算的时候有可能会数组越界。

代码：

import java.util.Scanner;

public class Main {
    static int x;
    static int n;
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        n=sc.nextInt();
        int sum=0;
        int[] a=new int[n];
        for (int i = 0; i < n; i++) {
            a[i]=sc.nextInt();
            sum+=a[i];
        }
        boolean[][] dp=new boolean[n][100005];
        dp[0][a[0]]=true;
        for (int i = 1; i < n; i++) {
            //开始是01背包问题，每个砝码的状态都是加上去，看能否凑出来这个数字
            for (int j = 1; j < dp[0].length; j++) {
                if (j==a[i]) {
                    dp[i][j]=true;
                }
                if (dp[i-1][j]) {
                    if (j+a[i]<dp[0].length) {
                        dp[i][j+a[i]]=true;
                    }
                    dp[i][j]=true;
                }
            }
        }
        for (int i = 0; i < dp.length; i++) {
            for (int j = 1; j < dp[0].length; j++) {
                if (j+a[i]<dp[0].length) {
                    if (dp[dp.length-1][j+a[i]]) {
                    //第二次循环是，假如j+a[i]这个数能够凑成，那么拿开这个a[i]砝码，
                    //，之后的重量也是能够凑出来的
                    dp[dp.length-1][j]=true;
                }
                }
                
            
            }
        }
        int ans=0;
        for (int i = 1; i < dp[0].length; i++) {
            if (dp[n-1][i]) {
                ans++;
            }
        }
        System.out.println(ans);
    }

}