题目描述
现有一组砝码,重量互不相等,分别为m1,m2,m3…mn;
每种砝码对应的数量为x1,x2,x3...xn。现在要用这些砝码去称物体的重量,问能称出多少中不同的重量。
注:
称重重量包括0
方法原型:public static int fama(int n, int[] weight, int[] nums)
输入包含多组测试数据。对于每组测试数据:第一行:n --- 砝码数(范围[1,10])第二行:m1 m2 m3 ... mn --- 每个砝码的重量(范围[1,2000])第三行:x1 x2 x3 .... xn --- 每个砝码的数量(范围[1,6])
利用给定的砝码可以称出的不同的重量数
21 22 1
5
import java.util.*;
public class Main{
public static void main(String[] args){
Scanner scan = new Scanner(System.in);
while(scan.hasNext()){
int n = scan.nextInt();
int[] m_weight = new int[n];//重量
int[] x_num = new int[n];//数量
for(int i = 0 ; i < n ; i++){
m_weight[i] = scan.nextInt();
}
for(int i = 0 ; i < n ; i++){
x_num[i] = scan.nextInt();
}
printNumOfDifferentWeight(n , m_weight , x_num);
}
scan.close();
}
private static void printNumOfDifferentWeight(int n , int[] m_weight , int[] x_num){
Set<Integer> set = new HashSet<Integer>();
//先求出第一种砝码可能称出的重量,个数可以为0~x_num[0] (即n+1)
for(int i = 0 ; i <= x_num[0] ; i++){
set.add(i * m_weight[0]);
}
//再在第一个砝码的基础上增加其他重量
//逐个增加砝码,更新集合set,i代表砝码的种类,从0开始
for(int i = 1 ; i < n ; i++){
//遍历第i个砝码的个数,从第二个砝码开始
//因为set无序集合并且没有get方法,所以将结果暂存于ArrayList中遍历
List<Integer> list = new ArrayList<Integer>(set);
//j遍历第i种砝码的个数
for(int j = 1 ; j <= x_num[i] ; j++){
for(int k = 0 ; k < list.size() ; k++){
set.add(list.get(k) + j * m_weight[i]);
}
}
}
System.out.println(set.size());
}
}