【简易贪心 LeetCode 12.7 每日一题】

LeetCode 12.7 每日一题 1775. 通过最少操作次数使数组的和相等

算法：贪心

1. 下面的讨论都以 $suma>sumb$ 来说，即 $a$ 是大数组，$b$ 是小数组。

2. 首先我们肯定能想到，要使得把两个数组的和变相等的操作次数最小，那就一定要将 $a$ 数组的最大数变小，或是 $b$ 数组的最小数变大，而变小变大的上界在本题中就是 $5$。因此我们可以有这样一个猜想：如果用上述方式，每一次修改只把 $a$ 的当前最大数改为 $1$，或 $b$ 的当前最小数改为 $6$，此时如果修改到了 $suma⩽sumb$，立即停止修改。这样就说明两个数组的和一定可以被修改到相等，并且此时就是最小的操作次数。

3. 我们来证明一下。首先，若改到最后 $suma=sumb$，那结论肯定成立，主要是 $suma<sumb$ 的情况。根据上面的描述，只有当 $suma>sumb$ 的时候，它才会每一次减少差值，中途如果有 $suma=sumb$，那就是第一种情况，所以它一定在最后一步有一个突变，从 $suma>sumb$ 到 $suma<sumb$，既然有如此大幅度的合法修改，我们不妨将最后一步的尺度缩小到刚好有 $suma=sumb$，这是绝对可行的，比如最后一步是把 $5$ 改为 $1$，我们完全可以把 $5$ 改为 $2$ 及以上，因此修改后 $suma⩽sumb$，就说明可以修改至 $suma=sumb$。接下来证为什么是最小操作次数。假设中间的某些步骤把改变的量变少了，那么结果一定不会变好，因为如果保持原方案不变，最后刚好 $suma=sumb$ ，那么把中间的改变量变小，一定会造成 $suma>sumb$，这样会让我们多操作一步；同理，对于原方案最后的结果是 $suma<sumb$ 的，如果中间步骤改变量变小，那么一定会导致 $suma⩾sumb$，有可能让我们多操作一步。综上，当前的贪心方案就是最优方案。

4. 而对于无法变成相等的数组，只有一种情况，就是小数组全变成 $6$，大数组全变成 $1$，这样的条件下大数组的和仍比小数组大，那就一定没有任何办法修改，因为只要小数组的上确界比大数组的下确界大或者相等，那说明二者进行适当的修改后一定存在交集。

5. 所以我们只需要开一个数组存下来两个数组能改变的量为 $0$ 到 $5$ 的有多少个（比如 $a$ 数组有 $7$ 个 $5$，那就说明他改变量为 $4$ 的个数有 $7$ 个，因为我们要将大数组的数都变成 $1$），然后从大到小（从 $5$ 到 $1$）枚举每个改变量，当 $suma-sumb⩽0$ 就退出循环。

#define all(x) x.begin(), x.end()

class Solution {
public:
    int minOperations(vector<int>& a, vector<int>& b) {
        int sa = accumulate(all(a), 0);
        int sb = accumulate(all(b), 0);
        if (sa == sb) return 0;
        if (sa < sb) return minOperations(b, a);
        int n = a.size(), m = b.size();
        if (6 * m < n) return -1;
        int d = sa - sb;
        vector<int> cnt(6);
        for (auto x: a) cnt[x - 1] ++ ;
        for (auto x: b) cnt[6 - x] ++ ;
        int res = 0;
        for (int i = 5; i && d > 0; i -- )
            while (cnt[i] && d > 0) {
                res ++ ;
                cnt[i] -- ;
                d -= i;
            }
        return res;
    }
};