堆排序使用练习--力扣6039. K 次增加后的最大乘积

力扣（leetcode）6039. K 次增加后的最大乘积，这题有些意思，

题目如下

给你一个非负整数数组 nums 和一个整数 k 。每次操作，你可以选择 nums 中 任一 元素并将它 增加 1 。

请你返回 至多 k 次操作后，能得到的 nums的 最大乘积 。由于答案可能很大，请你将答案对 109 + 7 取余后返回。

示例 1：

输入：nums = [0,4], k = 5
输出：20
解释：将第一个数增加 5 次。
得到 nums = [5, 4] ，乘积为 5 * 4 = 20 。
可以证明 20 是能得到的最大乘积，所以我们返回 20 。
存在其他增加 nums 的方法，也能得到最大乘积。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/maximum-product-after-k-increments
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

这个题目关键的地方在处理思路上，如果只加一个1，加在哪里会让乘积最大呢，

假如有 a < b, 其余的乘积为M，加在a上为 （a + 1）*b * M = (a*b + b) * M

加在b上为（b+1）*a * M = (a*b + a) * M

可以看出，加在较小的数上时，最优。

所以可以每次把最小的数加1，k次操作，可以使用堆排序来找到最小值。

这里每次加1其实比较慢，还可以再优化一下，直接把最小的数加到第二小的数字大小，

还有一个有意思的地方，(a + b) % c = (a%c + b%c) % c

(a* b) % c = (a%c * b) % c

这个问题可以变幻一下，如果给的不是数字k，给的是长度为k的数组，如何分配这k个数字呢？

假如有数字 a > b,     数组k中有 d2 > d1

比较 （a + d1）(b + d2) 和（a + d2）(b + d1) 

可以看出，把k数组里最大的数字分配到原数据中最小的数字上时，乘积最大

附C++堆排序处理code

class Solution {
public:
    //sort the root
    void down(vector<int>& nums, int len) {
        int p = 0;
        while (1) {
            int l = 2*p + 1;
            int r = 2*p + 2;
            if (p >= len) break;
            if (l >= len) break;
            if (r < len) {
                if (nums[p] < min(nums[l], nums[r])) break;
                if (nums[l] < nums[r]) {
                    swap(nums[l], nums[p]);
                    p = l;
                } else {
                    swap(nums[r], nums[p]);
                    p = r;
                }
            } else {
                if (nums[p] > nums[l]) {
                    swap(nums[l], nums[p]);
                    p = l;
                } else {
                    break;
                }
            }
        }
    }

    int maximumProduct(vector<int>& nums, int k) {
        //init, sort
        sort(nums.begin(), nums.end());
        int len = nums.size();
        for (int i = 0; i < k; i++) {
            nums[0]++;
            down(nums, len);
        }

        int m = 1000000000+7;
        long long t = 1;
        for (int i = 0; i < len; i++) {
            t *= nums[i];
            if (t > m) {
                t = t % m;
            }
        }

        return t;
    }
};