题目:
题目背景
出题是一件痛苦的事情!
相同的题目看多了也会有审美疲劳,于是我舍弃了大家所熟悉的 A+B Problem,改用 A-B 了哈哈!
题目描述
给出一串正整数数列以及一个正整数 C,要求计算出所有满足 A−B=C 的数对的个数(不同位置的数字一样的数对算不同的数对)。
输入格式
输入共两行。
第一行,两个正整数 N,C。
第二行,N 个正整数,作为要求处理的那串数。
输出格式
一行,表示该串正整数中包含的满足 A−B=C 的数对的个数。
输入输出样例
输入 #1复制
4 1 1 1 2 3
输出 #1复制
3
说明/提示
对于 75% 的数据,1≤N≤2000。
对于 100%的数据,1≤N≤2×1051,0≤ai<230,1≤C<230
2017/4/29 新添数据两组
代码
import bisect as b
a,c=map(int,input().split())
l=list(map(int,input().split()))
l.sort()
l1=[0 for i in range(a)]
for i in range(a):
l1[i] = l[i]-c
co = 0
for i in range(a):
d= b.bisect_left(l, l1[i])
s = b.bisect_right(l, l1[i])
co+=s-d
print(co)