存在重复元素 III(LeetCode)

题目

给你一个整数数组 nums 和两个整数 indexDiffvalueDiff

找出满足下述条件的下标对 (i, j)

  • i != j,
  • abs(i - j) <= indexDiff
  • abs(nums[i] - nums[j]) <= valueDiff

如果存在,返回 true 否则,返回false

解题

python 复制代码
"""
该算法的时间复杂度为 O(n log min(n, indexDiff)),其中 n 是数组的长度。
SortedList 的插入和删除操作均为 O(log k),k 为当前滑动窗口的大小
"""
from sortedcontainers import SortedList


def containsNearbyAlmostDuplicate(nums, indexDiff, valueDiff):
    # 有序列表,用于维护滑动窗口内的元素
    sorted_list = SortedList()

    for i, num in enumerate(nums):
        # 如果有滑动窗口大小限制,移除不在窗口内的元素
        if i > indexDiff:
            sorted_list.remove(nums[i - indexDiff - 1])

        # 查找范围内是否存在符合条件的元素
        pos1 = sorted_list.bisect_left(num - valueDiff)
        if pos1 < len(sorted_list) and sorted_list[pos1] <= num + valueDiff:
            return True

        # 将当前元素加入有序列表中
        sorted_list.add(num)

    return False


nums = [1, 2, 3, 1]
indexDiff = 3
valueDiff = 0
print(containsNearbyAlmostDuplicate(nums, indexDiff, valueDiff))  # 输出: True
相关推荐
九圣残炎3 分钟前
【从零开始的LeetCode-算法】3354. 使数组元素等于零
java·算法·leetcode
程序猿小柒44 分钟前
leetcode hot100【LeetCode 4.寻找两个正序数组的中位数】java实现
java·算法·leetcode
雨中rain1 小时前
贪心算法(1)
算法·贪心算法
不爱学习的YY酱1 小时前
【操作系统不挂科】<CPU调度(13)>选择题(带答案与解析)
java·linux·前端·算法·操作系统
平头哥在等你2 小时前
求一个3*3矩阵对角线元素之和
c语言·算法·矩阵
飞滕人生TYF2 小时前
动态规划 详解
算法·动态规划
_OLi_2 小时前
力扣 LeetCode 106. 从中序与后序遍历序列构造二叉树(Day9:二叉树)
数据结构·算法·leetcode
ahadee2 小时前
蓝桥杯每日真题 - 第18天
c语言·vscode·算法·蓝桥杯
我明天再来学Web渗透2 小时前
【SQL50】day 2
开发语言·数据结构·leetcode·面试
地平线开发者3 小时前
CPU& 内存加压工具 stress-ng 介绍
算法·自动驾驶