【题目描述】
给出一串数以及一个数字 C,要求计算出所有 A-B=C 的数对的个数。(不同位置的数字
一样的数对算不同的数对)
【输入格式】
第一行包括2个非负整数 N 和C(C≠0) ,中间用空格隔开。
第二行有 N个整数,中间用空格隔开,作为要求处理的那串数。
【输出格式】
输出一行,表示该串数中包含的所有满足 A-B=C的数对的个数。
【输入样例】dec.in
4 1
1 1 2 3
【输出样例】dec.out
3
【数据范围】
对于50%的数据,0 < N <= 200;
对于70%的数据,0 < N <= 2000;
对于100%的数据,0 < N <= 200000。
所有输入数据都在 C++的int范围内。
其实这道题开始我没想到用哈希,而是想直接通过确定A来确定B(跟哈希一样),但做了两个优化。
1:倒着枚举A,当a【i】<c时就一定找不到A了;
2:在剩下的数中找B时用个二分;
但这个代码只能得90分
代码如下:
#include<iostream>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
using namespace std;
int n,c,a[2000010];
int main()
{
scanf("%d%d",&n,&c);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
sort(a+1,a+n+1);
int ans=0,l,r,mid;
for(int i=n;i>=1;i--)
{
if(a[i]<=c)break;
l=1;
r=i-1;
while(l<=r)
{
mid=(l+r)>>1;
if(a[i]-a[mid]>c)
l=mid+1;
if(a[i]-a[mid]<c)
r=mid-1;
if(a[i]-a[mid]==c)
{
ans++;
int p=mid;
while(a[i]-a[mid]==c)
{
mid++;
if(a[i]-a[mid]==c)ans++;
}
while(a[i]-a[p]==c)
{
p--;
if(a[i]-a[p]==c)ans++;
}
break;
}
}
}
cout<<ans;
return 0;
}
后来考虑哈希。哈希思想很简单,这道题基本是裸哈希。哈希是将数字(或字母)按一定的运算规律储存(自己定,一般是除以一个大质数,好像要准确些?)并同时用一个数组(比如hash【】)记录用你的运算规律算完后产生的数字(记作j)的本来的样子(记作i),每次算出j后data【j】++(表示i出现了几次)。当然,可能会有不同的i算下来是同一个j,这时hash、data就有用了,当data【j】>=1时说明已经有过j了,当hash【j】(以前算出j的i)!=i(当前的i)时说明这两个数不相同,这时可以j++,存到下一个地方,如果还一样,再++。
代码:
#include<iostream>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
using namespace std;
int n,c,a[200010],ans=0,data[300007*2],hash[300007*2];
inline int loc(int x)
{
int j=x%300007;
while(data[j]>=1&&hash[j]!=x)j++;
return j;
}
int main()
{
scanf("%d%d",&n,&c);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
int j=loc(a[i]);
data[j]++;
hash[j]=a[i];
}
for(int i=1;i<=n;i++)
{
int j=loc(c+a[i]);
ans+=data[j];
}
cout<<ans;
return 0;
}