leetcode算法题--数组的最大美丽值

原题链接：https://leetcode.cn/problems/maximum-beauty-of-an-array-after-applying-operation/

这个题目需要转换一下思路，当一个数为x，那么子数组最大的元素会是x+2k，那么就是查找<= x+ 2k的元素位置。有两种做法，二分查找和滑动窗口。

1、滑动窗口

这里也有个技巧，枚举右端点

func maximumBeauty(nums []int, k int) int {
    sort.Ints(nums)
    res := 0
    i := 0
    for j, right := range nums {
        for right > nums[i] + 2 * k {
            i++
        }
        res = max(res, j - i + 1)
    }
    return res
}

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

    return b
}

2、二分查找

func maximumBeauty(nums []int, k int) int {
    sort.Ints(nums)
    n := len(nums)
    res := 0
    for i, num := range nums {
        left, right := i, n - 1
        for left <= right {
            target := num + 2 * k
            mid := (left + right) / 2
            if nums[mid] <= target {
                left = mid + 1
            } else {
                right = mid - 1
            }
        }
        res = max(res, right - i + 1)
    }
    return res
}

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