题解 | #称砝码#

称砝码

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

import java.util.HashSet;
import java.util.Scanner;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int weight[] = new int[n];
        int num[] = new int[n];
        for (int i = 0; i < n; i++) {
            weight[i] = in.nextInt();
        }

        for (int i = 0; i < n; i++) {
            num[i] = in.nextInt();
        }

        HashSet<Integer> set = new HashSet<Integer>();
        set.add(0);
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < num[i]; j++) {
                // 增加此砝码与不加的情况(不加则不修改set)
                HashSet<Integer> addSet = new HashSet<Integer>();
                for (Integer k : set) {
                    if (!set.contains(k + weight[i])) {
                        addSet.add(k + weight[i]);
                    }
                }
                set.addAll(addSet);
            }
        }
        System.out.println(set.size());
    }
}

一共N个砝码(n1 + n2 + ..+ nn)

每个砝码可选择加或不加,共2^n种选择

遍历砝码,添加则修改已有重量的set,不加则保持原有set,两个set合一继续遍历

全部评论

相关推荐

点赞 评论 收藏
分享
点赞 评论 收藏
分享
门口唉提是地铁杀:之前b站被一个游戏demo深深的吸引了。看up主页发现是个初创公司,而且还在招人,也是一天60。二面的时候要我做一个登录验证和传输文件两个微服务,做完要我推到github仓库,还要我加上jaeger和一堆运维工具做性能测试并且面试的时候投屏演示。我傻乎乎的做完以后人家跟我说一句现在暂时不招人,1分钱没拿到全是白干
你的秋招第一场笔试是哪家
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务