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

Posted 2020-10-15

题目背景

woshiren在洛谷刷题，感觉第一题：求两数的和(A+B Problem)太无聊了，于是增加了一题：A-B Problem，难倒了一群小朋友，哈哈。

题目描述

给出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]。

 

模拟

#include<map>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 200100
#define ll long long
using namespace std;
map<int,int>m;
ll n,c,ans,a[N],maxn;
ll read()
{
    ll x=0,f=1; char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘) x=x*10+ch-‘0‘,ch=getchar();
    return x*f;
}
int main()
{
    n=read(),c=read();
    for(int i=1;i<=n;i++)
     a[i]=read(),m[a[i]]++;
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++)
     ans+=m[a[i]+c];
    printf("%lld",ans);
    return 0;
}