python:洛谷P1102A-B数队

洛谷P1102 A-B数对是一道简单的算法题，题目要求计算出所有 A - B = C 的数对的个数。可以使用map类来解决这个问题。具体做法是，定义一个map数组，将这些数作为下标存入，将对应数字出现的次数作为这个数组中存的数，那么就可以直接枚举数组中的每一个数，通过这些数字，找到符合要求的B并在数组中找到下标为B所对应元素的出现次数，那么出现次数之和就是所有满足题目条件的总个数。具体代码如下：

n, c = map(int, input().split())
a = list(map(int, input().split()))
ds = {}
ans = 0
for i in range(n):
    b = a[i] - c
    if b in ds:
        ans += ds[b]
    if a[i] in ds:
        ds[a[i]] += 1
    else:
        ds[a[i]] = 1
print(ans)

其中，ds是一个字典，用于存储每个数字出现的次数。在遍历数组a时，先计算出当前数字对应的B，如果B在字典中出现过，则将其出现次数加入答案中。然后再将当前数字的出现次数加入字典中。最后输出答案即可。