前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 2233. K 次增加后的最大乘积(优先队列)

LeetCode 2233. K 次增加后的最大乘积(优先队列)

作者头像
Michael阿明
发布2022-05-10 09:51:29
2070
发布2022-05-10 09:51:29
举报
文章被收录于专栏:Michael阿明学习之路

文章目录

1. 题目

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

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

代码语言:javascript
复制
示例 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 <= 10^5
0 <= nums[i] <= 10^6

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

2. 解题

  • 从最小的数开始增加1,能获取最大的增长百分比,每次都是对新的数组进行此操作
  • 采用优先队列,小的优先,取出堆顶的+1,再放回
代码语言:javascript
复制
class Solution {
public:
    int maximumProduct(vector<int>& nums, int k) {
        priority_queue<int, vector<int>, greater<int>> q;
        for(auto num : nums)
            q.push(num);
        while(k--)
        {
            int num = q.top();
            q.pop();
            q.push(num+1);
        }
        int ans = 1;
        while(!q.empty())
        {
            ans = (1LL*ans*q.top())%1000000007;
            q.pop();
        }
        return ans;
    }
};

440 ms 88.6 MB C++

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022-05-02,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 文章目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档