开放日活动【C语言】

/*
【二分查找】
某部门开展Family Day开放日活动，其中有个从桶里取球的游戏，游戏规则如下:有N个容量一样的小桶等距排开，且每个小桶都默认装了数量不等的小球，
每个小桶装的小球数量记录在数组bucketBallNums中,游戏开始时，要求所有桶的小球总数不能超过SUM,
如果小球总数超过SUM，则需对所有的小桶统一设置一个容量最大值maxcapacity并需将超过容量最大值的小球拿出来，直至小桶里的小球数量小于maxcapacity;
请您根据输入的数据，计算从每个小桶里拿出的小球数量，
限制规则一:
所有所有小桶的小球总和小于SUM，则无需设置容量值，并且无需从小桶中拿球出来，返回结果[]
限制规则二:
如果所有小桶的小球总和大于SUM，则需设置容量最大值 maxcapacity并且需从小桶中拿球出来，返回从每个小桶拿出的小球数量组成的数组;
输入描述
第一行输入2个正整数，数字之间使用空格隔开，其中第一个数字表示SUM，第二个数字表示 bucketBalNums 数组长度
第二行输入N个正整数，数字之间使用空格隔开，表示bucketBallNums的每一项
输出描述
数组剩余小球。
示例一
输入
14 7
2 3 2 5 5 1 4
输出
[0,1,0,3,3,0,2]
说明
小球总数为22，sum=14，超出范围了，需从小桶取球，
maxCapacty=1，取出球后，桶里剩余小球总和为7，远小于14
maxCapacity=2，取出球后，桶里剩余小球总和为13
maxCapacity=3，取出球后，桶里剩余小球总和为17，大于14
因此 maxCapacity为2，每个小桶小球数量大于2的都需要拿出来;
示例二
输入
3 3
1 2 3
输出
[0,1,2]
说明
小球总数为6，SUM=3，超出范围了，需从小桶取球maxCapacity=1，则小球总数为3从0号桶取出0个球，从1号桶取出1个球，从2号桶取出2个球
示例三
输入
6 2
3 2
输出
[]
说明
小球总数为5，SUM=6，在范围内，无需从小桶取球
备注
1 <= bucketBallNums[i] <= 10^9
1 <= bucketBallNums.length = N <= 10^5
1 <= maxCapacity <= 10^9
1 <= SUM <= 10^9
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
 
#define MIN(a, b) ((a) < (b)) ? (a) : (b)
#define MAX(a, b) ((a) > (b)) ? (a) : (b)
 
int cmp (const void * a, const void * b) {
   return ( *(int*)a - *(int*)b );
}

int main(){
    int sum, n;
    scanf("%d %d", &sum, &n);
    int nums[n + 1];
    int const_nums[n + 1];//用来复原nums[]
    int original_nums[n];//不排序，仅用于输出结果
    nums[0] = 0;//给数组首元素填个0，我喜欢这样干
    const_nums[0] = 0;
    for(int i = 1; i <= n; i++){
        scanf("%d", &nums[i]);
        const_nums[i] = nums[i];
        original_nums[i - 1] = nums[i];
    }
    qsort(nums, n + 1, sizeof(int), cmp);//快排
    qsort(const_nums, n + 1, sizeof(int), cmp);

    //无需从小桶取球的情况
    int total = 0;
    for(int i = 0; i < n; i++){
        total += original_nums[i];
    }
    if(total <= sum){
        printf("[]");
        return 0;
    }

    //二分查找
    int left = nums[0];
    int right = nums[n];
    int result = 0;
    while(left < right){
        int change_sum = 0;
        int mid = (left + right) / 2;
        for(int i = 0; i <= n; i++){
            if(nums[i] > mid){
                nums[i] = mid;
            }
            change_sum += nums[i];
        }
        if(change_sum <= sum){
            result = MAX(result, mid);
            left = mid;
        }
        else {
            right = mid - 1;
        }
        for(int i = 0; i<= n; i++){
            nums[i] = const_nums[i];//复原nums[i]
        }
    }

    //输出结果
    printf("[");
    for(int i = 0; i < n; i++){
        if(i != 0){
            printf(",");
        }
        if(original_nums[i] <= result){
            original_nums[i] = 0;
        }
        else {
            original_nums[i] = original_nums[i] - result;
        }
        printf("%d", original_nums[i]);
    }
    printf("]");

    return 0;
}