洛谷-P1102 A-B 数对
洛谷-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\%\) 的数据，\(1 \leq N \leq 2000\)。
对于 \(100\%\) 的数据，\(1 \leq N \leq 2 \times 10^5\)。
保证所有输入数据都在 \(32\) 位带符号整数范围内。
2017/4/29 新添数据两组
C++代码
#include <cstdio>
#include <map>
using namespace std;
map <long long, long long> m;
int main() {
long long n, c, ans=0;
scanf("%lld%lld", &n, &c);
long long a[n];
for (int i=0; i<n; ++i)
scanf("%lld", &a[i]);
for (int i=0; i<n; ++i) {
++m[a[i]];
a[i] -= c;
}
for (int i=0; i<n; ++i)
ans += m[a[i]];
printf("%lld\n", ans);
return 0;
}
