差分数组学习：Leetcode 2779. 数组的最大美丽值

给你一个下标从 0 开始的整数数组 nums 和一个 非负 整数 k 。
在一步操作中，你可以执行下述指令：

在范围 [0, nums.length - 1] 中选择一个 此前没有选过 的下标 i 。
将 nums[i] 替换为范围 [nums[i] - k, nums[i] + k] 内的任一整数。

数组的 美丽值 定义为数组中由相等元素组成的最长子序列的长度。
对数组 nums 执行上述操作任意次后，返回数组可能取得的 最大 美丽值。

注意：你 只 能对每个下标执行 一次 此操作。
数组的 子序列 定义是：经由原数组删除一些元素（也可能不删除）得到的一个新数组，且在此过程中剩余元素的顺序不发生改变。

示例 1：
输入：nums = [4,6,1,2], k = 2
输出：3
解释：在这个示例中，我们执行下述操作：

• 选择下标 1 ，将其替换为 4（从范围 [4,8] 中选出），此时 nums = [4,4,1,2] 。
• 选择下标 3 ，将其替换为 4（从范围 [0,4] 中选出），此时 nums = [4,4,1,4] 。
  执行上述操作后，数组的美丽值是 3（子序列由下标 0 、1 、3 对应的元素组成）。
  可以证明 3 是我们可以得到的由相等元素组成的最长子序列长度。

示例 2：
输入：nums = [1,1,1,1], k = 10
输出：4
解释：在这个示例中，我们无需执行任何操作。
数组 nums 的美丽值是 4（整个数组）。

思路

自己做的思路是将每个数的范围想象为区间，然后区间排序，思路类似于交叠区间这样，用到了滑动窗口。

另一种很妙的思路是差分数组。本质上，我们要找的是，在所有的数字中，能被这些区间（也就是输入nums中的元素）命中的最大数字。但这样对于输入nums中每个元素的范围的每个数字都要更新其被命中的数量，这样复杂度会很高。这里用函数的思想：函数在区间内的值，等效于其导数的积分；同理，离散的数列，其值等于其差分的前n项和。因此，对于nums中一个元素，只需要更新差分的两项元素。

代码

这里给出差分数组方法的代码：

class Solution:
    def maximumBeauty(self, nums: List[int], k: int) -> int:
        m = max(nums)
        diff = [0] * (m + 2)
        for x in nums:
            diff[max(x - k, 0)] += 1
            diff[min(x + k + 1, m + 1)] -= 1
        res, count = 0, 0
        for x in diff:
            count += x
            res = max(res, count)
        return res