小马刷题记 - 2530 执行K次操作后的最大分数

题目

给你一个下标从 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

解答

解答一

逐次排序 + 贪心

class Solution {
        public long maxKelements(int[] nums, int k) {
            long result = 0;
            for(int i = 0; i < k; i++) {
                result += findMaxNum(nums);
            }
            return result;
        }

        private int findMaxNum(int[] nums) {
            int result = 0;
            int index = 0;
            for (int i = 0; i < nums.length; i++) {
                if (nums[i] > result) {
                    result = nums[i];
                    index = i;
                }
            }
            nums[index] = (result + 2)/3;
            return result;
        }
    }

解答二

优先队列 + 贪心

class Solution {
        public long maxKelements(int[] nums, int k) {
            PriorityQueue<Integer> q = new PriorityQueue<Integer>((a, b) -> b - a);
            for (int num : nums) {
                q.offer(num);
            }
            long ans = 0;
            for (int i = 0; i < k; ++i) {
                int x = q.poll();
                ans += x;
                q.offer((x + 2) / 3);
            }
            return ans;
        }
    }