LeetCode 1838. 最高频元素的频数（二分查找）

1. 题目

元素的 频数 是该元素在一个数组中出现的次数。

给你一个整数数组 nums 和一个整数 k 。
在一步操作中，你可以选择 nums 的一个下标，并将该下标对应元素的值增加 1 。

执行最多 k 次操作后，返回数组中最高频元素的 最大可能频数 。

示例 1：
输入：nums = [1,2,4], k = 5
输出：3
解释：对第一个元素执行 3 次递增操作，对第二个元素执 2 次递增操作，
此时 nums = [4,4,4] 。
4 是数组中最高频元素，频数是 3 。

示例 2：
输入：nums = [1,4,8,13], k = 5
输出：2
解释：存在多种最优解决方案：
- 对第一个元素执行 3 次递增操作，此时 nums = [4,4,8,13] 。
4 是数组中最高频元素，频数是 2 。
- 对第二个元素执行 4 次递增操作，此时 nums = [1,8,8,13] 。
8 是数组中最高频元素，频数是 2 。
- 对第三个元素执行 5 次递增操作，此时 nums = [1,4,13,13] 。
13 是数组中最高频元素，频数是 2 。

示例 3：
输入：nums = [3,9,6], k = 2
输出：1
 
提示：
1 <= nums.length <= 10^5
1 <= nums[i] <= 10^5
1 <= k <= 10^5

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

2. 解题

• 对数字进行排序（只有小的数能通过操作变成大的数），并求前缀和
• 二分求解最大频次
• 每次二分的时候，遍历一次前缀和数组，以某位置为最大频数的数字，能不能满足要求？根据结果，调整区间

类似题目：
LeetCode 1231. 分享巧克力（极小极大化 二分查找）

class Solution {
    int ans = 1;
public:
    int maxFrequency(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end());
        vector<long> presum(nums.size()+1);
        for(int i = 1; i <= nums.size(); ++i)
            presum[i] = presum[i-1]+nums[i-1];
        int l = 1, r = 1e5, mid;
        while(l <= r)
        {
            mid = l+((r-l)>>1);
            if(ok(nums, presum, mid, k))
            //在最大频数为mid的时候，能做到吗？
            {
                ans = mid;//能
                l = mid+1;
            }
            else
                r = mid-1;
        }
        return ans;
    }
    bool ok(vector<int>& nums, vector<long>& presum, long freq, int k)
    {
        for(int i = freq; i < presum.size(); i++)
        {
            if(presum[i]-presum[i-freq]+k >= nums[i-1]*freq)
            // 包含 i 位置在内的前面 freq 个数字都变成 i 位置处的数字，能做到吗
                return true;
        }
        return false;
    }
};

272 ms 84 MB C++

---

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！