值和下标之差都在给定的范围内——桶排序

80 阅读1分钟

image.png

  1. 桶排序思想 + 滑动窗口
  2. 按照元素的大小进行分桶,桶的大小为 t+1
  3. 如果两个元素属于同一个桶,两个元素显然符合条件
  4. 如果两个元素属于相邻的桶,则需要验证两个元素是否差值不超过 t
  5. 如果两个元素既不属于同一个桶,也不属于相邻桶,则必然不符合条件
func getID(x, w int) int {
    if x >= 0 {
        return x / w
    }
    return (x+1)/w - 1
}

func containsNearbyAlmostDuplicate(nums []int, k, t int) bool {
    mp := map[int]int{}
    for i, x := range nums {
        id := getID(x, t+1)
        if _, has := mp[id]; has {
            return true
        }
        if y, has := mp[id-1]; has && abs(x-y) <= t {
            return true
        }
        if y, has := mp[id+1]; has && abs(x-y) <= t {
            return true
        }
        mp[id] = x
        if i >= k {
            delete(mp, getID(nums[i-k], t+1))
        }
    }
    return false
}

func abs(x int) int {
    if x < 0 {
        return -x
    }
    return x
}