2021.7.12提高B组模拟1T1 好元素（hash）

Posted 2021-08-11 SSL_LKJ

好元素

题目大意

给出A数组
若$$a[m]+a[n]+a[p]=a[i]$$
$$（1<=m,n,p<i）（m,n,p可以相同）$$
即 a[i] 为好元素
求好元素数量

输入样例

1.

2
1 3

2.

6
1 2 3 5 7 10

3.

3
-1 2 0

输出样例

1.

1

2.

4

3.

1

解题思路

这题我们可以先转化一下思路

改为

$$a[m]+a[n]=a[i]-a[p]$$

然后我们可以发现时间复杂度O（n²）

接着处理3*10⁹的数组内存

我们不难可想到hash表

最后这道题就AC了

AC代码

#include<cstdio>
using namespace std;
const int hash=25000005;//5000*5000
int n,ans,a[5005],ha[hash+5];
int gethash(int x)//求hash
{
	int i=0,o=(x%hash+hash)%hash;//防止负数
	while(i<hash&&ha[(o+i)%hash]!=2147483647&&ha[(o+i)%hash]!=x)i++;
	return (o+i)%hash;
}
int main()
{
	freopen("good.in","r",stdin);
	freopen("good.out","w",stdout);
	scanf("%d",&n);
	for(int i=0;i<hash;i++)ha[i]=2147483647;//初值
	for(int i=1;i<=n;i++)
	{
	 	scanf("%d",&a[i]);
		for(int j=1;j<i;j++)
		 if(ha[gethash(a[i]-a[j])]==a[i]-a[j])//转化后的hash
		 {
		 	ans++;
		 	break;
		 }
		for(int j=1;j<=i;j++)ha[gethash(a[i]+a[j])]=a[i]+a[j];//插入
	}
	printf("%d",ans);
	return 0;
} 

谢谢