称砝码

描述
现有一组砝码，重量互不相等，分别为m1、m2……mn；他们可取的最大数量分别为x1、x2……xn。现在要用这些砝码去称物体的重量，问能称出多少中不同的重量。
注：
称重重量包括0
要对输入数据进行校验
方法原型：public static int fama(int n, int[] weight, int[] nums)
知识点 字符串,循环,函数,指针,枚举,位运算,结构体,联合体,文件操作,递归
运行时间限制 10M
内存限制 128
输入
int n：n表示有多少组重量不同的砝码，1<=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<>
int[] weight：表示n组砝码的重量，1<=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<>
int[] num：表示n组砝码的最大数量，1<=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<>
输出
利用给定的砝码可以称出的不同的重量数
样例输入 2 1 2 2 1
样例输出 5

解题思路：
动态规划求解，子问题就是前i个砝码的重量总数为Q[i],子问题，那么Q[i]一定可以再Q[i-1]的基础上计算出来，由底往上，在Q[i-1]的基础上，Q[i-1]分别增加k个第i个砝码，并且在添加的过程中去除重复的重量。
假设：weight[N] 表示N个不同砝码的重量，
num[N]表示N个不同砝码的个数
那么Q[i]=(Q[i-1]+k*weight[i])-添加过程中已经出现的重量 。k的范围从0到num[i]。
flags[j]作为一个标记数组，标记重量j是否出现过了，出现过就标记为1。其中flags[0]作为一个临时变量来作为每添加一个砝码遍历结束时结束的位置，最后统计标记数组中1的个数即可。

import java.util.*;

public class Main{

    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        int N=sc.nextInt();

        int[] weights=new int[N];
        int[] nums=new int[N];
        for (int i=0;i<N;i++) {
            weights[i]=sc.nextInt();
        }

        for (int i=0;i<N;i++) {
            nums[i]=sc.nextInt();
        }
        System.out.println(fama(N,weights,nums));
    }

    public static int fama(int N,int[] weight,int[] num){
        int[] flags=new int[4001];  //一开始提交的时候，没有计算出4001这个值，而是用乘积，提交错误，原因未知。
        int total=0;
        int Num=0;

        //flags[0]用于每添加一个砝码时遍历的结束位置,即最大的那个重量边界。
        flags[0]=weight[0]*num[0];
        // System.out.println(flags[0]);

        //添加一个砝码时的重量做标记
        for (int i=1;i<=num[0];i++) {
            flags[weight[0]*i]=1;
        }


        for (int i=1;i<N;i++) {     //子问题的个数，从1到N每添加一个砝码变成一个新的子问题，直至变成母问题
            int m=flags[0];         //m变量作为前一个子问题的结束时重量
            for (int k=1;k<=num[i];k++) {   //砝码的个数增加计算新的重量
                for (int j=0;j<=m;j++) {    //m作为前一个子问题的右边界，增加砝码后新的重量从0,1,2,3,4...m 在此基础上+k*weight[i]
                    if (j+weight[i]*k>4000) {   //判断是否越界
                        break;
                    }
                    //　当处理第j个砝码，质量为wj时，有下列推导公式：
                    // flags[j-kw]==>d[j]=1(当且仅当j-kw已经存在时，才能获得j)
                    if ((flags[j]==1)&&flags[j+k*weight[i]]!=1||j==0) { //后一个的重量必须在新的重量存在前提下才能获得，否则不会获得新重量，或者没有砝码
                        flags[j+k*weight[i]]=1; //标记数组
                        total=j+k*weight[i];    //获得增加砝码后遍历结束时的高值边界
                    }
                }
            }
            flags[0]=total; //增加一个砝码后将新的右边界赋值，然后接着循环增加下一个砝码的情况。
        }

        for (int i=1;i<=4000;i++ ) {
            if (flags[i]==1) {
                // System.out.println(flags[i]);
                Num++;
            }
        }
        return Num+1;
    }


}

参考动态规划链接：
华为OJ称砝码1
华为OJ称砝码2
华为OJ称砝码3
动态规划和母函数法称砝码