编程语言
首页 > 编程语言> > Leetcode 1838. 最高频元素的频数-C++

Leetcode 1838. 最高频元素的频数-C++

作者:互联网

题目描述:

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

给你一个整数数组 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

 

操作步骤:

第一步: 对数据进行排序

第二步: 将范围边界分为左边界l和右边界r,循环边界r,如果nums[r] * (r - l + 1) > sum + k; 

将左边界向右移动一位,sum -= nums[l++], sum的值减少

第三步: 使用max获得最大的r - l + 1的结果

 

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

class Solution {
public:
    int maxFrequency(vector<int>& nums, int k) {
        sort(begin(nums), end(nums));
        int sum = 0;
        int l = 0;
        int ans = 0;
        for(int r = 0; r < nums.size(); r++)
        {
            sum += nums[r];
            while(l < r && static_cast<long>(sum + k) < (nums[r] * (r - l + 1))){
                sum -= nums[l++];
            }
            ans = max(ans, r - l + 1);
        }
        return ans;
    }
};

int main()
{
    Solution s;
    vector<int>nums = {1, 2, 4};
    int value = s.maxFrequency(nums, 5);
    std::cout << "value:" << value << std::endl;
}

 

标签:频数,13,nums,int,1838,元素,C++,sum,Leetcode
来源: https://www.cnblogs.com/my-love-is-python/p/15004507.html