题目描述
给出一串正整数数列以及一个正整数 C,要求计算出所有满足 A−B=C 的数对的个数(不同位置的数字一样的数对算不同的数对)。
输入格式
输入共两行。
第一行,两个正整数 N,C。
第二行,N 个正整数,作为要求处理的那串数。
输出格式
一行,表示该串正整数中包含的满足 A−B=C 的数对的个数。
输入输出样例
输入 #1复制
4 1 1 1 2 3
输出 #1复制
3
说明/提示
对于 75%75% 的数据,1≤N≤2000。
对于 100%100% 的数据,1≤N≤2×10^5,0≤ai<2^30,1≤C<2^30。
思路:利用map函数;
#include<iostream>
#include<map>
using namespace std;
map<long,long>b;//利用map函数;
int main()
{
int n;
long c,ans=0,a[200001];
cin>>n>>c;
for(int i=1;i<=n;i++)
{
cin>>a[i];
b[a[i]]++;
a[i]-=c;
}//将A数组出现的次数统计;同时减去差值;
for(int i=1;i<=n;i++)
ans+=b[a[i]];//将A数组扫一遍;
cout<<ans<<endl;
return 0;
}