牛客2021暑期训练1-H-Hash Function

牛客2021暑期训练1-H-Hash Function

题目链接

题意

给定n个互不相同的数，找一个最小模域，使得它们在模域下不同，n<=500000,a[i]<=500000

题解

卷积，FFT/NTT
对任意两个数 ai%mod == aj%mod 可以转换为 | a[i]-a[j] |%mod==0
要求所有差值复杂度 n^2，此时就要用到卷积在 nlogn 的复杂度下求出

推荐两个写的好的FFT/NTT文章
倾心讲解 FFT 多项式与快速傅里叶变换 && 迭代模板带注释
小学生都能看懂的FFT！！！

FFT/NTT练手

那这卷积怎么求差呢
一个序列 2 3 4
取相反数 -2 -3 -4
转换为两数相加了，是不是像多项式的次数，多项式相乘即为次数相加，而FFT/NTT不能出现负数，因此将负数加上 N=500000
卷积完后枚举模数判断

代码

#include<bits/stdc++.h>
#define DB double
using namespace std;
const int N=500000,M=2000009;
const DB Pi=acos(-1.0);
int n=1,m,len,p;
int r[M],vis[M];
struct node
{
	DB x,y;
	node() {}
	node(DB u,DB v) {x=u; y=v;}
	node operator * (const node &u) const {return node(x*u.x-y*u.y,x*u.y+y*u.x);}
	node operator + (const node &u) const {return node(x+u.x,y+u.y);}
	node operator - (const node &u) const {return node(x-u.x,y-u.y);}
}a[M],b[M];
void FFT(node *c,DB flag)
{
	for(int i=0;i<=n;i++)
		if(i<r[i])
			swap(c[i],c[r[i]]);
	for(int i=1;i<n;i<<=1)
	{
		node T(cos(Pi/i),flag*sin(Pi/i));
		for(int k=0;k<n;k+=(i<<1))
		{
			node t(1,0);
			for(int j=0;j<i;j++,t=t*T)
			{
				node nx=c[k+j],ny=t*c[k+j+i];
				c[k+j]=nx+ny;
				c[k+j+i]=nx-ny;
			}
		}
	}
}
int main()
{
	cin>>p;
	for(int i=1;i<=p;i++)
	{
		int x;
		cin>>x;
		a[x].x=1;
		b[N-x].x=1;
	}
	len=N<<1;
	while(n<=len) n<<=1,m++;
	for(int i=0;i<=n;i++)
		r[i]=(r[i>>1]>>1)|((i&1)<<(m-1));
	FFT(a,1);
	FFT(b,1);
	for(int i=0;i<=n;i++)
		a[i]=a[i]*b[i];
	FFT(a,-1);
	for(int i=N;i<=N*2;i++)
		vis[i-N]=a[i].x/n+0.5;
	for(int i=p;i<=N+1;i++)
	{
		int flag=0;
		for(int j=i;j<=N;j+=i)
			flag|=vis[j];
		if(!flag)
		{
			cout<<i;
			break;
		}
	}
	return 0;
}