LeetCode 1838. 最高频元素的频数--前缀和+双指针

1838. 最高频元素的频数

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

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

题解

这题需要下思维转换，首先对数组排序，我们可以暴力枚举，nums[j]和前面的数字，把j左边的元素都加到nums[j]，那么需要加nums[j]*(j-i+1)-(sum[j]-sum[i-1])次，其中sum[i]表示0到i的前缀和。

于是再利用双指针快速找到符合条件的i和j，然后保存最大的差值作为答案。

AC代码

class Solution {
public:
    typedef long long ll;
    ll sum[100010];
    int maxFrequency(vector<int>& nums, int k) 
    {
        memset(sum,0,sizeof(sum));
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size();i++)
        sum[i+1]=sum[i]+nums[i];
        int mxlen=0;
        int s1=1,s2=1;
        while(s1<=nums.size()&&s2<=nums.size())//双指针
        {
            if((ll)nums[s2-1]*(s2-s1+1)-(sum[s2]-sum[s1-1])<=k)
            {
                mxlen=max(mxlen,s2-s1+1);
                s2++;
            }
            else 
            {
                s1++;
            }
        }
        return mxlen;
    }
};