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

思路:

使用滑动窗口法是最容易想到的,注意不一定是从第一个开始的

代码:

func maxFrequency(nums []int, k int) int {
    sort.Ints(nums)
    n := len(nums)
    maxCnt := 1
    sum :=0
    for l,r := 0,1;r<n;r++{
        sum += (r-l)*(nums[r]-nums[r-1])
        for sum>k{
            sum -= nums[r]-nums[l]
            l++
        }
        maxCnt = max(maxCnt, r-l+1)
    }
    return maxCnt
}

func max(a,b int) int{
    if a>b{
        return a
    }
    return b
}

代码效率:

执行用时172 ms, 在所有 Go 提交中击败了100.00%的用户
内存消耗:9 MB, 在所有 Go 提交中击败了19.23%的用户


本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!