Leetcode 3347. Maximum Frequency of an Element After Performing Operations II

• Leetcode 3347. Maximum Frequency of an Element After Performing Operations II
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3347. Maximum Frequency of an Element After Performing Operations II

1. 解题思路

这一题思路上其实还好，我们首先将所有的数字进行排序，然后考察每一个数字作为频率最高的数字的情况下可以获得多大的出现次数，然后将其返回即可。

需要注意的是，对于每一个数，我们需要考虑三个值，即：

• 这个值本身作为目标值
• 这个值作为目标值的左边界
• 这个值作为目标值的右边界

而对于任意给定的目标值$n$，我们要考察其能够获得的最大的出现频次，我们只需要考察$[n-k,n+k]$这个区间内一共有多少数字以及有多少恰好等于$n$的值即可，我们可以用一个二分搜索快速得到我们的答案。

2. 代码实现

给出python代码实现如下：

class Solution:
    def maxFrequency(self, nums: List[int], k: int, numOperations: int) -> int:
        cnt = Counter(nums)
        nums = sorted(nums)
        
        @lru_cache(None)
        def count(num):
            left = bisect.bisect_left(nums, num-k)
            right = bisect.bisect_right(nums, num+k)
            return min(right-left-cnt[num], numOperations) + cnt[num]
        
        ans = 1
        for num in nums:
            ans = max(ans, count(num-k), count(num), count(num+k))
            
        return ans

提交代码评测得到：耗时1390ms，占用内存65.2MB。