【洛谷P1102】A-B数对【二分】【哈希】

题目描述

出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 A+B Problem，改用 A-B 了哈哈！

好吧，题目是这样的：给出一串数以及一个数字 CC，要求计算出所有 A - B = CA−B=C 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行，两个整数 N, CN,C。

第二行，NN 个整数，作为要求处理的那串数。

输出格式

一行，表示该串数中包含的满足 A - B = CA−B=C 的数对的个数。

输入输出样例

输入 #1

4 1
1 1 2 3

输出 #1

3

说明/提示

对于 75% 的数据，1≤N≤2000。

对于 100% 的数据，1≤N≤2×10⁵ 。

保证所有输入数据都在 32 位带符号整数范围内。

分析

这题暴力STL二分应该是比较简单就可以做出来的。排序之后upper_bound-lower_bound累加一下即可。
哈希做法的话一开始我调了很久都没有搞出来，也不知道是哪里出问题。后面不知改了什么很玄学的东西就很突然地过了。其实跟前面的题没啥区别，输入之后把这个数直接定位（代码中的dw函数），然后累加这个数出现的个数。进入哈希表就随便用一个比较大的质数作为“模数”，正常的哈希累加出答案。注意最后是减法得到的答案才累加到ans里面。

上代码

二分法

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n,c,a[200000];
int main()
{
	cin>>n>>c;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	sort(a+1,a+n+1);
	long long ans=0;
	for(int i=1;i<=n;i++)
	{
		ans+=upper_bound(a+1,a+n+1,a[i]-c)-lower_bound(a+1,a+n+1,a[i]-c);
	}
	cout<<ans;
	return 0;
}

哈希

#include<iostream>
#include<cstdio>
#include<algorithm> 
#include<cstring>
#include<cmath>
#define p 4000037
using namespace std;
long long a[p],b[200010],c[p];
long long n,C,x,qm,ans;
long long hash(long long x)
{
	return x%p;
}
long long dw(long long x)
{
	long long j=0;
	qm=hash(abs(x));
	while(a[hash(j+qm)]!=x&&c[hash(j+qm)]!=0)
	{
		j++;
	} 
	return hash(j+qm);
}
int main()
{
	scanf("%lld%lld",&n,&C);
	for(int i=1;i<=n;i++)
	{
	 	scanf("%lld",&b[i]);
	 	a[dw(b[i])]=b[i];      //一个存数，一个存数出现的次数
	 	c[dw(b[i])]++;
	}
	for(int i=1; i<=n; i++)
	{
		ans+=c[dw(b[i]-C)%p];
	} 
    cout<<ans;
    return 0;
}