map + 思维

P1102 A-B 数对

给出一串数以及一个数字 C，要求计算出所有 A−B=C的数对的个数（不同位置的数字一样的数对算不同的数对）。
输入格式

输入共两行。

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

第二行，N 个整数，作为要求处理的那串数。
输出格式

一行，表示该串数中包含的满足 A−B=C的数对的个数。
输入输出样例
输入

4 1
1 1 2 3

输出

3

思路

①将A-B=C转换成A-C=B
②将A数组每个元素出现的次数统计起来，用map映射，
③最后将A数组每次减一个C，再将A数组扫一遍，将所有映射的次数和加起来就是答案

题解

#include<bits/stdc++.h>
using namespace std;
#define N 2000005
typedef long long int ll;
ll a[N],n,c,sum = 0;
map<ll,ll>m;

int main(){
	cin >> n >> c;
	for(int i=1;i<=n;i++){
		cin >> a[i];
		m[a[i]]++;
		a[i]-=c;
	}
	for(int i=1;i<=n;i++)
		sum += m[a[i]];
	cout << sum << endl;
	return 0;
}

原题连接：https://www.luogu.com.cn/problem/P1102