1838. 最高频元素的频数

旧城等待， 2022-08-30 11:19 3阅读 0赞

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

给你一个整数数组 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 <= 105  
1 <= nums\[i\] <= 105  
1 <= k <= 105

class Solution {
    public:
        int maxFrequency(vector<int>& nums, int k) {
            sort(nums.begin(),nums.end());
            long long s[100001]={0};
            for(int i = 1 ;i<=nums.size();i++)
            {
                s[i] = s[i-1]+nums[i-1];
            }
            int l = 0,r = nums.size();
            while(l<r)
            {
                int mid = (l+r+1)/2;
                int flag=0;
                for(int i = mid;i<=nums.size();i++)
                {
                    if(k>=(long long)nums[i-1]*mid-s[i]+s[i-mid])
                    {
                        flag = 1;
                        break;
                    }
                }
                if(flag) 
                {
                    l=mid;
                }
                else r = mid-1;
            }
            return l;
        }
    };