洛谷P1102 A-B 数对

题目描述
出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 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% 的数据， 20001≤N≤2000。
对于 100% 的数据，1≤N≤2×10^5。
保证所有输入数据都在 32 位带符号整数范围内。
2017/4/29 新添数据两组

满足A - B = C 的数对也满足A - C = B
这时我们可以把问题转化为 B是否在原数组中且B出现了几次

#include<iostream>
#include<map>
using namespace std;
typedef long long ll;
ll n,c;
map<ll, ll>mp;
ll ans=0;
int a[200001];
int main()
{
	cin >> n>>c;
	mp.clear();
	for (int i = 0; i < n; i++)
	{
		cin >> a[i];
		mp[a[i]]++;   //原数组中B的出现次数
		a[i] -= c;    //A-C就是我们要找的B
	}
	for (int i = 0; i < n; i++)
	{
		ans += mp[a[i]];  //遍历加上B在原数组出现的次数即可
	}
	cout << ans << endl;
	return 0;
}