A-B数对（C++）题解

题干：

给出一串正整数数列以及一个正整数 CC，要求计算出所有满足 A - B = CA−B=C 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入要求：

输入共两行。

第一行，两个正整数 N,CN,C。

第二行，NN 个正整数，作为要求处理的那串数。

输出要求：

一行，表示该串正整数中包含的满足 A - B = CA−B=C 的数对的个数。

输入样本：

4 1
1 1 2 3

输出样本：

3 

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e6;
int a[N];
map<int,int>s;
int main()
{
	int n,c; scanf("%d %d",&n,&c);
    long long count=0;
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
		s[a[i]]++;
		a[i]=a[i]-c;
	}
	for(int i=0;i<n;i++){
	    count=count+s[a[i]];
	}
	printf("%lld",count);
}