其他分享
首页 > 其他分享> > 力扣(leetcode) 1838. 最高频元素的频数 (滑动窗口法)

力扣(leetcode) 1838. 最高频元素的频数 (滑动窗口法)

作者:互联网

题目在这:https://leetcode-cn.com/problems/frequency-of-the-most-frequent-element/

题目分析:

简单的说一下这个题目的意思,就是给你一个数组,比如[1,2,3]
再给你一个正数,比如 K = 3。
你可以将数组内的任意数字加1,操作3次,使数组内相同数字出现的次数最多。
比如上述例子,将 1 加两次 等于3 。2加一次等于3.此时 ‘加1’ 操作进行了3次, 小于等于题目中给的K。且数组中为【3,3,3】 ”3“ 出现了三次,是最多的。
最后返回出现次数最多的数字的出现次数。返回 ‘3’ 。

思路分析:

这道题的数量级都在 105 比较大,使用暴力必定会超时。

首先要知道,本题中的操作只能加,不能减。
比如 [2,4] 我们只能将2加到4 ,不能将4减到2.
所以,当K无限大时,要找数组里的某个数字最大频数,一定是将所有数字全部加到该数组的最大值。
比如 [1,4,7] 将1和4全都加到7,此时数字 7 的出现次数 3次,为最大。

所以我们使用排序+滑动窗口来解决本题。

设置两个指针分别指向窗口的头和尾。

当窗口内的数字全部加到最大所需操作数不大于K为符合条件。

1. 窗口内符合条件:右指针后移,扩大窗口。计算新进来的数字所需操作数,更新总的所使用的操作数。
2. 窗口内不符合条件:左指针后移,缩小窗口。计算吐出去的数字所需操作数。更新总的所使用的的操作数。

完整代码:

 def maxFrequency(self, nums: List[int], k: int) -> int:
 
        l = 0
        r = 1
        nums.sort()
        res =1 # 记录最大频数
        temp = 0 # 记录加一次数

        while r < len(nums):
            temp += (r - l) * (nums[r] - nums[r - 1]) # 更新当前操作数

            while temp > k:

                temp -= nums[r] - nums[l]
                l += 1
            res = max(res, r - l + 1)
            r += 1
        return res

解释一下代码里temp变量计算思想。

例如 nums = [1,3,5,7] K=5

开始时 左指针L 指向1,右指针R指向3。
将1 加到 3 所需操作数为2次。 小于5。 R指针后移

第二次while时。 数组为 [3,3,5,7] temp为2。
此时R指针指向5 L指针依旧指向第一个3.
计算此时的操作数temp,为上一次将1加到3所需操作数加上 本次将两个3都变成R所指的5所需的操作数。r-l 计算左右指针之间的数字,nums[r] - nums[r - 1] 计算 右指针和右指针前一位的差值。

要明白,每次循环时,除了右指针,其他窗口内的数字都是一样的,因为都在前面改变过了。

所以,当窗口内增加值了。temp = temp +(r - l) * (nums[r] - nums[r - 1])
当窗口内的值减少了。temp = temp - nums[r] - nums[l]

标签:操作数,窗口,数字,temp,nums,1838,力扣,leetcode,指针
来源: https://blog.csdn.net/qq_38737428/article/details/118957834