給你一個下標從 0 開始的整數陣列 nums 和一個整數 k 。
你可以對陣列執行 至多 k 次操作：
從陣列中選擇一個下標 i ，將 nums[i] 增加 或者 減少 1 。
最終陣列的頻率分數定義為陣列中眾數的 頻率 。
請你返回你可以得到的 最大 頻率分數。
眾數指的是陣列中出現次數最多的數。一個元素的頻率指的是陣列中這個元素的出現次數。
示例 1：
輸入：nums = [1,2,6,4], k = 3
輸出：3
解釋：我們可以對陣列執行以下操作：
- 選擇 i = 0 ，將 nums[0] 增加 1 。得到陣列 [2,2,6,4] 。
- 選擇 i = 3 ，將 nums[3] 減少 1 ，得到陣列 [2,2,6,3] 。
- 選擇 i = 3 ，將 nums[3] 減少 1 ，得到陣列 [2,2,6,2] 。
元素 2 是最終陣列中的眾數，出現了 3 次，所以頻率分數為 3 。
3 是所有可行方案裡的最大頻率分數。
class Solution:
def maxFrequencyScore(self, nums: List[int], k: int) -> int:
nums.sort()
n = len(nums)
pre = [n for n in nums]
pre.insert(0, 0)
for i in range(1, n + 1):
pre[i] += pre[i - 1]
def count(left, right):
mid = (left + right) // 2
return (
pre[right + 1]
- pre[mid + 1]
- (right - mid) * nums[mid]
+ nums[mid] * (mid - left)
- (pre[mid] - pre[left])
)
left = 0
res = 0
for right in range(n):
while count(left, right) > k:
left += 1
res = max(res, right - left + 1)
return res