洛谷-P1102 A-B 数对

洛谷-P1102 A-B 数对

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

• 题目描述
• 输入格式
• 输出格式
• 输入输出样例
• 说明/提示
• C++代码

---

题目描述

出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 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 imes 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
", ans);
    return 0;
}