华为OJ 初级：称砝码

最新推荐文章于 2020-09-09 09:56:14 发布

Wang_Y_

最新推荐文章于 2020-09-09 09:56:14 发布

阅读量381

点赞数

分类专栏：【华为OJ】

本文链接：https://blog.csdn.net/wanggg2760/article/details/52072923

版权

【华为OJ】专栏收录该内容

66 篇文章 1 订阅

订阅专栏

描述	现有一组砝码，重量互不相等，分别为m1、m2……mn；他们可取的最大数量分别为x1、x2……xn。现在要用这些砝码去称物体的重量，问能称出多少中不同的重量。注：称重重量包括0 要对输入数据进行校验方法原型：public static int fama(int n, int[] weight, int[] nums)
知识点	字符串,循环,函数,指针,枚举,位运算,结构体,联合体,文件操作,递归
运行时间限制	10M
内存限制	128
输入	int n：n表示有多少组重量不同的砝码，1<=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<><=n<=10<> int[] weight：表示n组砝码的重量，1<=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<><=mi<=10<> int[] num：表示n组砝码的最大数量，1<=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<><=xi<=10<>
输出	利用给定的砝码可以称出的不同的重量数
样例输入	2 1 2 2 1
样例输出	5

import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;
//复制的别人的方法
public class Main {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt(); // 种类数
		int[] weight = new int[n];// 砝码重量数组
		int[] nums = new int[n]; // 砝码数量数组
		Set<Integer> result = new HashSet<Integer>();// 保存可能的重量数的set
		for (int i = 0; i < n; i++) {
			weight[i] = sc.nextInt();
		}
		for (int i = 0; i < n; i++) {
			nums[i] = sc.nextInt();
		}
		sc.close();
		getWeightSet(0, 0, weight, nums, result);
		System.out.println(result.size());
	}

	public static void getWeightSet(int curWeight, int curIndex, int[] weight,
			int[] nums, Set<Integer> result) {
		// 递归结束条件，当已遍历完所有砝码种类则将重量加入到集合中
		if (curIndex >= weight.length) {
			result.add(curWeight);
			return;
		}
		// 递归转移方式，对当前索引的每一种可能数量递归调用该方法
		for (int i = 0; i <= nums[curIndex]; i++) {
			getWeightSet(curWeight + weight[curIndex] * i, curIndex + 1,
					weight, nums, result);
		}
	}
}