题解 | #称砝码#

称砝码

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());
        }
    }
}

全部评论

相关推荐

点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务