HJ41 称砝码
描述
现有 n 种砝码,重量互不相等,分别为 m1,m2,m3…mn ;
每种砝码对应的数量为 x1,x2,x3…xn 。现在要用这些砝码去称物体的重量(放在同一侧),问能称出多少种不同的重量。
注:
称重重量包括 0
数据范围:每组输入数据满足 1≤n≤10 1≤n≤10 , 1≤mi≤2000 1≤mi≤2000 , 1≤xi≤10 1≤xi≤10
输入描述:
对于每组测试数据:
第一行:n — 砝码的种数(范围[1,10])
第二行:m1 m2 m3 … mn — 每种砝码的重量(范围[1,2000])
第三行:x1 x2 x3 … xn — 每种砝码对应的数量(范围[1,10])
输出描述:
利用给定的砝码可以称出的不同的重量数
题解
HashSet
import java.util.Scanner;
import java.util.Set;
import java.util.HashSet;
import java.util.ArrayList;
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
// 注意 hasNext 和 hasNextLine 的区别
while (in.hasNextInt()) {
// 砝码的种数 n
int n = Integer.valueOf(in.nextLine());
// 每种砝码的重量 互不相等
String str1 = in.nextLine();
// 每种砝码的数量
String str2 = in.nextLine();
// 输出可以称出的不同数量的重量数
int res = getRes(n, str1, str2);
System.out.println(res);
}
}
public static int getRes(int n, String a, String b) {
String[] strA = a.split(" ");// 重量
String[] strB = b.split(" ");// 种数
// 存放可能表示的重量
Set <Integer> set = new HashSet<>();
set.add(0);
for (int i = 0; i < n; i++) {
// 遍历砝码
ArrayList <Integer> list = new ArrayList<>(set);// 取当前所有的结果
for (int j = 1; j <= Integer.valueOf(strB[i]); j++) {
// 遍历砝码个数
for (int k = 0; k < list.size(); k++) {
// 遍历前面表示的重量
set.add(list.get(k) + Integer.valueOf(strA[i]) * j);
}
}
}
// 表示的重量 与 种数
return set.size();
}
}