数对个数(哈希基础题)

 
【题目描述】 
给出一串数以及一个数字 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;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值