Python每日一题——差分计数

额嗯146

于 2025-03-18 21:12:39 发布

阅读量130

点赞数 2

CC 4.0 BY-SA版权

分类专栏： python数据结构文章标签： python 数据结构哈希表

本文链接：https://blog.youkuaiyun.com/Sechoice/article/details/146352518

python数据结构专栏收录该内容

3 篇文章

订阅专栏

题目

在这里插入图片描述

思路

1.暴力枚举
直接枚举所有下标(i,j)，检查是否满足ai-aj，时间复杂度O(n^2)
对于2*10^6，显然会超时.
2.哈希表
记录每个数出现的次数
检查ai+x是否存在

代码

n,x=map(int,input().split())
a = list(map(int,input().split()))
count=0
T = {}
for i in a:
    T[i]=T.get(i,0)+1
for i in T.keys():
    if (i+x) in T:
        count+=T[i]*T[i+x]
print(count)