洛谷砝码称重

学编程的小程

于 2025-03-21 11:09:47 发布

阅读量1.3k

点赞数 51

分类专栏：算法篇章-捷报文章标签：算法动态规划 leetcode

本文链接：https://blog.csdn.net/weixin_74888502/article/details/146416220

版权

算法篇章-捷报专栏收录该内容

40 篇文章

订阅专栏

洛谷砝码称重

题目描述

你有一架天平和 ( N ) 个砝码，这 ( N ) 个砝码重量依次是 ( W_1, W_2, \dots, W_N )。请你计算一共可以称出多少种不同的重量？注意砝码可以放在天平两边。

示例

示例 1

输入:

3
1 4 6

输出:

解释:
能称出的 10 种重量是：1、2、3、4、5、6、7、9、10、11。

思路分析

问题核心

我们需要计算使用 ( N ) 个砝码，通过将它们放在天平的两边，可以称出多少种不同的重量。

思路拆解

初始化：
- 使用一个集合 set 来记录所有可能的重量，初始时集合中只有 0。
遍历砝码：
- 对于每个砝码，更新集合 set，将当前砝码的重量加到集合中的每个元素上，或者从集合中的每个元素中减去当前砝码的重量。
统计结果：
- 最终集合 set 的大小减去 1（去掉 0）就是可以称出的不同重量的数量。

代码段

import java.util.ArrayList;
import java.util.Arrays;
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[] dp = new int[n];
        for (int i = 0; i < n; i++) {
            dp[i] = sc.nextInt();
        }
        sc.close();
        Arrays.sort(dp);

        Set<Integer> set = new HashSet<>();
        set.add(0);
        for (int i = 0; i < dp.length; i++) {
            ArrayList<Integer> list = new ArrayList<>(set);
            for (int j : list) {
                set.add(j + dp[i]);
                set.add(Math.abs(j - dp[i]));
            }
        }

        System.out.println(set.size() - 1);
    }
}

在这里插入图片描述

代码逐行讲解

输入处理：

Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int[] dp = new int[n];
for (int i = 0; i < n; i++) {
    dp[i] = sc.nextInt();
}
sc.close();
Arrays.sort(dp);

使用 Scanner 读取输入的整数 n 和 n 个砝码的重量，并存储在数组 dp 中。
对数组 dp 进行排序。

初始化集合：
```
Set<Integer> set = new HashSet<>();
set.add(0);
```
- 初始化一个集合 set，并添加初始值 0。

遍历砝码：

for (int i = 0; i < dp.length; i++) {
    ArrayList<Integer> list = new ArrayList<>(set);
    for (int j : list) {
        set.add(j + dp[i]);
        set.add(Math.abs(j - dp[i]));
    }
}

对于每个砝码，遍历集合 set 中的每个元素，将当前砝码的重量加到元素上，或者从元素中减去当前砝码的重量，并将结果添加到集合 set 中。

输出结果：
```
System.out.println(set.size() - 1);
```
- 输出集合 set 的大小减去 1（去掉 0）作为结果。

复杂度分析

时间复杂度

遍历每个砝码的时间复杂度为 O(N)。
对于每个砝码，遍历集合 set 的时间复杂度为 O(M)，其中 ( M ) 是集合 set 的大小。
因此，总时间复杂度为 O(N * M)。

空间复杂度

使用了一个集合 set 来存储所有可能的重量，空间复杂度为 O(M)，其中 ( M ) 是集合 set 的大小。

总结的知识点

输入处理：
- 使用 Scanner 读取输入，并处理整数和数组。
集合操作：
- 使用 HashSet 来存储和更新所有可能的重量。
遍历与更新：
- 使用 for 循环遍历砝码和集合 set，并根据砝码的重量更新集合。
数学计算：
- 使用 Math.abs 计算绝对值，确保重量的非负性。

整合

import java.util.ArrayList;
import java.util.Arrays;
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[] dp = new int[n];
        for (int i = 0; i < n; i++) {
            dp[i] = sc.nextInt();
        }
        sc.close();
        Arrays.sort(dp);

        Set<Integer> set = new HashSet<>();
        set.add(0);
        for (int i = 0; i < dp.length; i++) {
            ArrayList<Integer> list = new ArrayList<>(set);
            for (int j : list) {
                set.add(j + dp[i]);
                set.add(Math.abs(j - dp[i]));
            }
        }

        System.out.println(set.size() - 1);
    }
}