C语言-贪心算法-背包问题-珠子放入背包

1.问题描述

你有 k 个背包。给你一个下标从 0 开始的整数数组 weights ，其中 weights[i] 是第 i 个珠子的重量。同时给你整数 k 。

请你按照如下规则将所有的珠子放进 k 个背包。

• 没有背包是空的。
• 如果第 i 个珠子和第 j 个珠子在同一个背包里，那么下标在 i 到 j 之间的所有珠子都必须在这同一个背包中。
• 如果一个背包有下标从 i 到 j 的所有珠子，那么这个背包的价格是 weights[i] + weights[j] 。

一个珠子分配方案的 分数 是所有 k 个背包的价格之和。

请你返回所有分配方案中，最大分数 与 最小分数 的 差值 为多少。

2.示例

示例 1：

输入：weights = [1,3,5,1], k = 2
输出：4
解释：
分配方案 [1],[3,5,1] 得到最小得分 (1+1) + (3+1) = 6 。
分配方案 [1,3],[5,1] 得到最大得分 (1+3) + (5+1) = 10 。
所以差值为 10 - 6 = 4 。

示例 2：

输入：weights = [1, 3], k = 2
输出：0
解释：唯一的分配方案为 [1],[3] 。
最大最小得分相等，所以返回 0 。

3.源码

#define  _CRT_SECURE_NO_WARNINGS 
#include <stdio.h>
#include <stdlib.h>
void swap(int* a, int* b)
{
    int temp = *a;
    *a = *b;
    *b = temp;
}
// 按照重量从小到大排序珠子
void sortWeights(int* weights, int n)
{
    for (int i = 0; i < n - 1; i++)
    {
        for (int j = 0; j < n - i - 1; j++) 
        {
            if (weights[j] > weights[j + 1]) 
            {
                swap(&weights[j], &weights[j + 1]);
            }
        }
    }
}
int calculateMaxDiff(int k, int n, int* weights)
{
    int max_diff = 0;
    sortWeights(weights, n);

    for (int i = 0; i <= n - k; i++)
    {
        int current_diff = weights[i + k ] - weights[i];
        if (current_diff > max_diff) {
            max_diff = current_diff;
        }
    }
    return max_diff;
}

int main()
{
    int k, n;
    printf("请输入珠子数量和背包数量：");
    scanf("%d %d", &n, &k);
    int* weights = (int*)malloc(n * sizeof(int));
    printf("请输入每个珠子的重量：");
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &weights[i]);
    }
    int maxDiff = calculateMaxDiff(k, n, weights);
    printf("最大分数与最小分数的差值为：%d\n", maxDiff);
    free(weights);
    return 0;
}

4.运行结果