leetcode 2233. K 次增加后的最大乘积

最新推荐文章于 2024-11-04 12:20:00 发布

我很忙2010

最新推荐文章于 2024-11-04 12:20:00 发布

阅读量991

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/qq_27060423/article/details/124231022

版权

LeetCode 专栏收录该内容

713 篇文章 10 订阅

订阅专栏

该博客讨论了一种算法问题，其中目标是在不超过k次操作的情况下，通过增加非负整数数组nums的元素值来最大化数组的乘积。示例展示了如何使用优先队列（最小堆）实现这一优化策略，最终返回经过操作后的最大乘积。该问题涉及到数值优化和数据结构的运用。

摘要由CSDN通过智能技术生成

给你一个非负整数数组 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 的方法，也能得到最大乘积。

示例 2：

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

提示：

1 <= nums.length, k <= 105
0 <= nums[i] <= 106

C++

class Solution {
public:
    int maximumProduct(vector<int>& nums, int k) {
        priority_queue<int,vector<int>,greater<int>> que;
        for(auto num:nums) {
            que.push(num);
        }
        while(k) {
            int num=que.top();
            que.pop();
            num++;
            que.push(num);
            k--;
        }
        long res=1;
        int mod=1000000007;
        while(!que.empty()) {
            res=((long)res*(long)que.top())%mod;
            que.pop();
        }
        return res;
    }
};

java

class Solution {
    public int maximumProduct(int[] nums, int k) {
        PriorityQueue<Integer> que = new PriorityQueue<>();
        for (int num : nums) {
            que.add(num);
        }
        while (k > 0) {
            int num = que.poll();
            num++;
            que.add(num);
            k--;
        }
        long res = 1;
        int mod = 1000000007;
        while (!que.isEmpty()) {
            res = res * ((long) que.poll()) % mod;
        }
        return (int)res;
    }
}