题目背景
出题是一件痛苦的事情!
相同的题目看多了也会有审美疲劳,于是我舍弃了大家所熟悉的 A+B Problem,改用 A-B 了哈哈!
题目描述
给出一串正整数数列以及一个正整数 C,要求计算出所有满足 A - B = C 的数对的个数(不同位置的数字一样的数对算不同的数对)。
输入格式
输入共两行。
第一行,两个正整数 N,C。
第二行,N 个正整数,作为要求处理的那串数。
输出格式
一行,表示该串正整数中包含的满足 A - B = C 的数对的个数。
样例 #1
样例输入 #1
4 1
1 1 2 3
样例输出 #1
3
提示
对于 75% 的数据,1 <= N <= 2000。
对于 100% 的数据,1 <= N <= 2 x 10^5,0 <= a_i <2^30,1 <= C < 2^30。
这道题是求两个数相减为C的数对的数量,我们不妨换个思路,A与C相减为B。因为B是在所给数列中出现的,所以我们可以用哈希表来存储每个数出现的次数去求解答案。
看代码。
#include <iostream>
#include <map>
using namespace std;
typedef long long ll;
const int N = 200020;
map<ll, ll> m;
ll a[N];
int main()
{
int n;
ll c;
ll ans = 0;
cin >> n >> c;
for (int i = 0; i < n; i++)
{
cin >> a[i];
m[a[i]]++;
a[i] -= c;
}
for (int i = 0; i < n; i++)
{
ans += m[a[i]];
}
cout << ans;
}