题解 | #称砝码#

称砝码

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

#include <stdio.h>

int main() {
    int n = 0;                           // 砝码种类数n
    scanf("%d", &n);
    int* weights = (int*)malloc(4 * n);
    int* cnts = (int*)malloc(4 * n);
    for (int i = 0; i < n; ++i) {
        scanf("%d", weights + i);                 // 砝码重量数组weights
    }
    for (int i = 0; i < n; ++i) {
        scanf("%d", cnts + i);                 // 砝码个数数组cnts
    }
    int max = 0;
    for (int i = 0; i < n; ++i) {
        max += weights[i] *
               cnts[i];                // 计算最大能称重的重量,哈希表的数组大小 max
    }
    char* hash = (char*)malloc(max + 1);                   // 哈希表,判断是否重复
    int maxSize = 1;                        // 计算可能的最多组合数
    for (int i = 0; i < n; ++i) maxSize *= cnts[i];
    int* sums = (int*)malloc(4 *maxSize);                   // sums用来存放不重复的前面元素的计算和
    sums[0] = 0;                          // 首元素为0
    int num = 1;                         // num 记录数量
    for (int i = 0; i < n; ++i) {
        int cnt = cnts[i];                  // 当前遍历到的砝码数量和重量
        int weight = weights[i];
        int size = num;                   // 遍历该砝码之前,已经存在的和数组的大小size
        for (int j = 1; j <= cnt; ++j) {         // 用sums数组的数逐个和 k 个 weight 进行组合相加,得到 sum
            for (int k = 0; k < size; ++k) {
                int sum = sums[k] + j * weight;
                // printf("sum %d k %d\n",sum,k);
                if (hash[sum] != '1') {       // sum不重复,则用哈希表记录,并存储到sums值,个数num++
                    sums[num++] = sum;
                    hash[sum] = '1';
                }
            }
        }
    }
    printf("%d", num);
    return 0;
}

解决使用给定的砝码和数量来计算可以称出的不同重量的数量。代码中使用了一个哈希表来避免重复计算相同的重量,并且使用了一个数组 sums 来存储所有可能的重量。

全部评论

相关推荐

11-15 17:19
湖南大学 Java
成果成果成果果:这是哪个公司的hr,这么离谱吗,我没见过用性别卡技术岗的,身边女性同学拿大厂offer的比比皆是
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务