题解 | #称砝码#
称砝码
https://www.nowcoder.com/practice/f9a4c19050fc477e9e27eb75f3bfd49c
用 HashSet 记录所有可能的重量,需要注意的是,不要忘了只使用第 i 种砝码的情况;以及不能在同一个 Set 中修改,因为上一次修改会影响本次修改。
或者,您也可以选择一个砝码一个砝码地添加,这样做需要注意对于同一个 Set,一般不允许一边遍历一边增加,可以选择复制 Set 或者使用 CopyOnWriteArraySet。
import java.util.*; // 注意类名必须为 Main, 不要有任何 package xxx 信息 public class Main { public static void main(String[] args) { Scanner in = new Scanner(System.in); // 注意 hasNext 和 hasNextLine 的区别 while (in.hasNextInt()) { // 注意 while 处理多个 case int n = in.nextInt(); int[] m = new int[n]; int[] x = new int[n]; for (int i = 0; i < n; i++) { m[i] = in.nextInt(); } for (int i = 0; i < n; i++) { x[i] = in.nextInt(); } Set<Integer> weights = new HashSet<>(); for (int i = 0; i < n; i++) { // 使用第 i 种砝码能获得的其他重量 HashSet<Integer> addSet = new HashSet<>(); for (int j = 0; j <= x[i]; j++) { // 添加若干个 m[i] for (Integer weight : weights) { addSet.add(weight + m[i] * j); } // 只使用 m[i] 的情况 addSet.add(m[i] * j); } weights.addAll(addSet); } System.out.println(weights.size()); } } }