题解 | #称砝码#

称砝码

https://www.nowcoder.com/practice/f9a4c19050fc477e9e27eb75f3bfd49c

import java.util.*;
public class Main {
    /**
       * 现有n种砝码,重量互不相等,分别为 m1,m2,m3…mn ;
       * 每种砝码对应的数量为 x1,x2,x3...xn 。现在要用这些砝码去称物体的重量(放在同一侧),问能称出多少种不同的重量。
       * 注:
       * 称重重量包括 0
       * 对于每组测试数据:
       * 第一行:n --- 砝码的种数(范围[1,10])
       * 第二行:m1 m2 m3 ... mn --- 每种砝码的重量(范围[1,2000])
       * 第三行:x1 x2 x3 .... xn --- 每种砝码对应的数量(范围[1,10])
       */

    /**
    * 首先根据输入顺序,将砝码用数字序列表示,例如 2个 1g 和 1个 2g,就用 1 1 2 的序列表示
    * 使用 Set 表示加入当前砝码之前能产生的重量种类
    * set 初始化为 {0};
    * 当第一个 1g 砝码放入时,则 set 中需要插入原先 set 中的所有元素 +1g后的结果,即 {0, 0+1}
    * 当第二个 1g 加入时,则 set 会插入 {0+1, 1+1},所以 set 最终变为{0, 1, 2}
    * 重复上述步骤加入所有砝码
    * 最后 set 的大小即为能产生的重量种类
    */
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        while (in.hasNext()) {
            int n = in.nextInt();
            int[] ms = new int[n];
            int[] xs = new int[n];
            for (int i = 0; i < n; i++) {
                ms[i] = in.nextInt();
            }
            for (int i = 0; i < n; i++) {
                xs[i] = in.nextInt();
            }

            List<Integer> list = new ArrayList<>();
            for (int i = 0; i < n; i++) {//所有砝码放到一个list中
                for (int j = 0; j < xs[i]; j++) {
                    list.add(ms[i]);
                }
            }

            Set<Integer> set = new HashSet<>();
            set.add(0);//0算一种重量
            for (Integer i : list) {
                List<Integer> tempList = new ArrayList<>(set.size());
                for (Integer val :
                        set) {//Set表示加入当前砝码之前能产生的重量种类
                    tempList.add(val + i);//新产生的重量种类
                }
                set.addAll(tempList);//去重
            }
            System.out.println(set.size());
        }
    }
}

动态规划和回溯算法 实在看不懂 只有偷偷学一种更好理解的办法

全部评论

相关推荐

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