洛谷 P2421 A-B数对（增强版）

题目描述

给出N 个从小到大排好序的整数，一个差值C，要求在这N个整数中找两个数A 和B，使得A-B=C，问这样的方案有多少种？

例如：N=5，C=2，5 个整数是：2 2 4 8 10。答案是3。具体方案：第3 个数减第1 个数；第3 个数减第2 个数；第5 个数减第4 个数。

输入输出格式

输入格式：

第一行2 个正整数：N,C。

第二行N 个整数：已经有序。注意：可能有相同的。

输出格式：

一个整数，表示该串数中包含的所有满足A-B=C 的数对的方案数。

输入输出样例

输入样例#1： 

4 1
1 1 2 2

输出样例#1： 

4

说明

对于50% 的数据：N 的范围是[1…1,000]。

对于另外50% 的数据：N 的范围是[1…100,000]。

对于100% 的数据：C 的范围是[1…1,000,000,000]，N 个整数中每个数的范围是：[0…1,000,000,000]。

 

Solution：

　　普及的题目，很水的思路。因为统计y-x=c的次数，所以就只要统计x+c的值出现了多少次，那么用m[a[i]]统计a[i]这个数出现的次数，这样答案就是m[a[i]+c]的累加和，因为数据较大，所以m数组需要hash，于是果断用STL的map映射，解决。

代码：

 

// luogu-judger-enable-o2
#include<bits/stdc++.h>
#define il inline
#define debug printf("%d %s\n",__LINE__,__FUNCTION__)
using namespace std;
map<int,int>m;
int n,c,a[500200];
il int gi()
{
    int a=0;char x=getchar();bool f=0;
    while((x<'0'||x>'9')&&x!='-')x=getchar();
    if(x=='-')x=getchar(),f=1;
    while(x>='0'&&x<='9')a=a*10+x-48,x=getchar();
    return f?-a:a;
}
int main()
{
    n=gi();c=gi();
    int ans=0;
    for(int i=1;i<=n;i++)a[i]=gi(),m[a[i]]++;
    for(int i=1;i<=n;i++)ans+=m[a[i]+c];
    //for(int i=1;i<=n;i++)ans+=m[a];
    cout<<ans;
    return 0;
}