洛谷 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映射,解决。

代码:

 1 // luogu-judger-enable-o2
 2 #include<bits/stdc++.h>
 3 #define il inline
 4 #define debug printf("%d %s
",__LINE__,__FUNCTION__)
 5 using namespace std;
 6 map<int,int>m;
 7 int n,c,a[500200];
 8 il int gi()
 9 {
10     int a=0;char x=getchar();bool f=0;
11     while((x<'0'||x>'9')&&x!='-')x=getchar();
12     if(x=='-')x=getchar(),f=1;
13     while(x>='0'&&x<='9')a=a*10+x-48,x=getchar();
14     return f?-a:a;
15 }
16 int main()
17 {
18     n=gi();c=gi();
19     int ans=0;
20     for(int i=1;i<=n;i++)a[i]=gi(),m[a[i]]++;
21     for(int i=1;i<=n;i++)ans+=m[a[i]+c];
22     //for(int i=1;i<=n;i++)ans+=m[a];
23     cout<<ans;
24     return 0;
25 }
原文地址:https://www.cnblogs.com/five20/p/8278387.html