HJ41 称砝码

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