存在重复元素 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
相关推荐
用户50408278583923 分钟前
1. RAG 权威指南:从本地实现到生产级优化的全面实践
算法
Python×CATIA工业智造1 小时前
详细页智能解析算法:洞悉海量页面数据的核心技术
爬虫·算法·pycharm
Swift社区2 小时前
Swift 解 LeetCode 321:拼接两个数组中的最大数,贪心 + 合并全解析
开发语言·leetcode·swift
无聊的小坏坏2 小时前
力扣 239 题:滑动窗口最大值的两种高效解法
c++·算法·leetcode
黎明smaly2 小时前
【排序】插入排序
c语言·开发语言·数据结构·c++·算法·排序算法
YuTaoShao3 小时前
【LeetCode 热题 100】206. 反转链表——(解法一)值翻转
算法·leetcode·链表
YuTaoShao3 小时前
【LeetCode 热题 100】142. 环形链表 II——快慢指针
java·算法·leetcode·链表
CCF_NOI.3 小时前
(普及−)B3629 吃冰棍——二分/模拟
数据结构·c++·算法
运器1234 小时前
【一起来学AI大模型】支持向量机(SVM):核心算法深度解析
大数据·人工智能·算法·机器学习·支持向量机·ai·ai编程
Zedthm4 小时前
LeetCode1004. 最大连续1的个数 III
java·算法·leetcode