Skip to content
目录

2530. 执行 K 次操作后的最大分数

难度:中等

地址:https://leetcode.cn/problems/maximal-score-after-applying-k-operations/description/

给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。你的 起始分数0

在一步 操作 中:

选出一个满足 0 <= i < nums.length 的下标 i , 将你的 分数 增加 nums[i] ,并且 将 nums[i] 替换为 ceil(nums[i] / 3) 。 返回在 恰好 执行 k 次操作后,你可能获得的最大分数。

向上取整函数 ceil(val) 的结果是大于或等于 val 的最小整数。

示例 1:

:nums = [10,10,10,10,10], k = 5

:50

:对数组中每个元素执行一次操作。最后分数是 10 + 10 + 10 + 10 + 10 = 50 。

示例 2:

:nums = [1,10,3,3,3], k = 3

:17

:可以执行下述操作: 第 1 步操作:选中 i = 1 ,nums 变为 [1,4,3,3,3] 。分数增加 10 。 第 2 步操作:选中 i = 1 ,nums 变为 [1,2,3,3,3] 。分数增加 4 。 第 3 步操作:选中 i = 2 ,nums 变为 [1,1,1,3,3] 。分数增加 3 。 最后分数是 10 + 4 + 3 = 17 。

提示:

  • 1 <= nums.length, k <= 10^5
  • 1 <= nums[i] <= 10^9

题解:

【官方】方法一:贪心 + 优先队列

js
/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var maxKelements = function (nums, k) {
    q = new MaxPriorityQueue();
    let ans = 0;
    for (const num of nums) {
        q.enqueue(num);
    }
    for (let i = 0; i < k; i++) {
        const x = q.dequeue().element;
        ans += x;
        q.enqueue(Math.ceil(x / 3));
    }
    return ans;
};

【我的答案】超出时间限制:

js
/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var maxKelements = function (nums, k) {
    let score = 0;
    for (let i = 0; i < k; i++) {
        let max = Math.max(...nums);
        let idx = nums.indexOf(max);
        score += nums[idx];
        nums[idx] = Math.ceil(nums[idx] / 3);
    }
    return score;
};