A-B数对 (hash映射)

题目大意:

第一行输入N,C
第二行输入n个数字
输出,求A - B = C的数对个数

样例

4 1
1 1 2 3

输出

3

思路:用STL容器map,map<num, times>,建立一个数字出现的次数的映射,题目要求是A - B = C,我们将其转换成A - C = B,并且将a数组依次减去C,最后再将A扫描一遍,将所有映射的次数加起来就是结果res

代码

#include <map>
#include <cstdio>
#include <iostream>

using namespace std;

const int N = 200010;
typedef long long LL;
LL a[N];
map<LL, LL> mp;

int main()
{
    int n;
    LL res = 0, c;

    scanf("%d%d", &n, &c);
    for(int i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
        mp[a[i]]++;
        a[i] -= c;
    }

    for(int i = 1; i <= n; i++) res += mp[a[i]];

    printf("%d", res);
    system("pause");
    return 0;
}

这个思路很巧妙,原理想了很久~但是理解的还不是很透彻,个人理解:假设a数组中减去数之后变为b数组,那么若想要原数组a中的产生A - B = C数对,那么b数组中的数也一定存在于原来的a数组,因此可以直接遍历a数组进行计数

按照题给样例
为了更好理解 a[]数组挨个减去C之后变成b[]数组
a[1] = 1; C = 1 那么a[]数组中还需要一个元素 0 才能与 1 构成数对,但是 0 并不存在,因此b[1] = 0,即mp[0] = 0;
a[2] = 1; C = 1 同理b[2] = 0,mp[0] = 0.
a[3] = 2; C = 1 那么a[]数组中需要一个 1 才能与 2 构成数对,刚好 1 有两个,b[3] = 1,即mp[1] = 2;
a[4] = 3; C = 1 那么a[]数组需要一个 2 才能与 3 构成数对, 刚好 2 有一个,b[4] = 2,即mp[2] = 1;
因此res = mp[0] + mp[0] + mp[1] + mp[2] = 3

posted @ 2020-07-29 18:44  Xxaj5  阅读(285)  评论(0编辑  收藏  举报