洛谷 P1102 A−B数对

题目描述

出题是一件痛苦的事情！

题目看多了也有审美疲劳，于是我舍弃了大家所熟悉的 A+BA+BA+B ProblemProblemProblem ，改用 A−BA-BA−B 了哈哈！

好吧，题目是这样的：给出一串数以及一个数字 CCC ，要求计算出所有 A−B=CA-B=CA−B=C 的数对的个数。（不同位置的数字一样的数对算不同的数对）

输入输出格式

输入格式：

第一行包括 222 个非负整数 NNN 和 CCC ，中间用空格隔开。

第二行有 NNN 个整数，中间用空格隔开，作为要求处理的那串数。

输出格式：

输出一行，表示该串数中包含的所有满足 A−B=CA-B=CA−B=C 的数对的个数。

输入输出样例

输入样例#1： 复制

4 1
1 1 2 3

输出样例#1： 复制

3

说明

对于 73%73\%73% 的数据， N≤2000N \le 2000N≤2000 ；

对于 100%100\%100% 的数据， N≤200000N \le 200000N≤200000 。

所有输入数据都在longint范围内。

2017/4/29新添数据两组

直接枚举是N^2算法，会超时。

换一种思路，枚举b，计算b+c的个数。

这是桶的思想，需要写离散化，博主太懒了，map大法好。

#include<map>
#include<cstdio>
#include<iostream>
using namespace std;
const int N=;
int n,c,a[N];
long long ans;
map<int,int>mp;
int main()
{
    scanf("%d%d",&n,&c);
    for(int i=;i<=n;i++)
    {
        scanf("%d",a+i);
        mp[a[i]]++;
    }
    for(int i=;i<=n;i++)
        ans+=mp[a[i]+c];
    printf("%lld\n",ans);
    return ;
}