题解 | #称砝码#
称砝码
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());
}
}
}
查看25道真题和解析
拼多多集团-PDD公司福利 817人发布