【洛谷 P1102】A-B 数对 题解（映射）

A-B 数对

题目背景

出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 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\le N\le 2000$。

对于 $100\%$ 的数据，$1\le N\le 2\times 10^5$，$0\le a_i<2^{30}$,$1\le C<2^{30}$。

2017/4/29 新添数据两组

思路

A - B = C 转化为 A = B + C。用map记录数字出现的次数。
若在map中，查找到存在A = B + C ，计数器加上A出现的次数与B出现的次数的乘积。

AC代码

#include 
#include 
#define AUTHOR "HEX9CF"
using namespace std;

int n, c;
long long cnt = 0;
map<int, long long> m;

int main()
{
    cin >> n >> c;
    for (int i = 0; i < n; i++)
    {
        int t;
        cin >> t;
        m[t]++;
    }
    map<int, long long>::iterator it = m.begin(), fit;
    for (; it != m.end(); it++)
    {
        // cout << it->first << " " << it->second << endl;
        if (m.end() != (fit = m.find(it->first + c)))
        {
            cnt += it->second * fit->second;
        }
    }
    cout << cnt << endl;
    return 0;
}