洛谷 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≤N≤2000。

对于 100% 的数据，1≤N≤2×105，0≤ai<230，1≤C<230。

2017/4/29 新添数据两组

思路：

换一种思路，A-B=C变成A-C=B。也就是A出现的次数乘以（A-C）出现的次数。

代码如下：

#include <iostream>
#include <unordered_map>
using namespace std;
typedef long long ll;

int main() 
{
    ll N, C; 
    cin >> N >> C; 
    unordered_map<ll, ll> k;
    ll ans = 0; 

    for (ll i = 0; i < N; i++) 
	{
        ll t;
        cin >> t;
        k[t]++;
    }

    for (auto it = k.begin(); it != k.end(); it++) 
	{
		ll key = it -> first;
		if (k.find(key - C) != k.end())//查找key-C，是否存在 
        ans += k[key] * k[key - C]; // 直接访问并累加，如果key - C不存在则结果为0
    }

    cout << ans << endl;

    return 0;
}