2021牛客暑期多校训练营1 H Hash Function FFT\NTT

传送门

文章目录

题意:

给你一个数组 a a a,你需要找一个最小的模数 x x x,使得 a a a中每个数都模上 x x x之后互不相同。
n ≤ 5 e 5 , a i ≤ 5 e 5 , a i ! = a j n\le5e5,a_i\le5e5,a_i!=a_j n5e5,ai5e5,ai!=aj

思路:

考虑两个数   m o d   x \bmod x modx相等意味着什么,写成表达式的形式就是 a i + x ∗ k = a j a_i+x*k=a_j ai+xk=aj,也就说我们如果知道每两个数之间的差,让后将差的因子筛掉,剩下的最小的数即为答案。
显然可以用 f f t fft fft加速上面的求每两个数之间的差的过程,由于存在负数,所以加一个偏移量即可,让后对于每个数遍历一下以他为因数的数看看是否被标记即可,这样可以避免根号的分解过程。
复杂度 O ( n l o g n ) O(nlogn) O(nlogn)

**// Problem: Hash Function
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/11166/H
// Memory Limit: 524288 MB
// Time Limit: 4000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4.1,sse4.2,avx,avx2,popcnt,tune=native")
//#pragma GCC optimize(2)
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#include<random>
#include<cassert>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid ((tr[u].l+tr[u].r)>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
using namespace std;

//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); }
//void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); }
//void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;

const int N=6000010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6,PI=acos(-1);

int n;
int rev[N],p[N];
int bit,limit;
int c[N];

struct Complex {
	double x,y;
	Complex operator + (const Complex& t) const { return {x+t.x,y+t.y}; }
	Complex operator - (const Complex& t) const { return {x-t.x,y-t.y}; }
	Complex operator * (const Complex& t) const { return {x*t.x-y*t.y,x*t.y+y*t.x}; } 
}a[N],b[N];

void fft(Complex a[],int inv) {
	for(int i=0;i<limit;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
	for(int mid=1;mid<limit;mid<<=1) {
		Complex w1=Complex({cos(PI/mid),inv*sin(PI/mid)});
		for(int i=0;i<limit;i+=mid*2) {
			Complex wk=Complex({1,0});
			for(int j=0;j<mid;j++,wk=wk*w1) {
				Complex x=a[i+j],y=wk*a[i+j+mid];
				a[i+j]=x+y; a[i+j+mid]=x-y;
			}
		}
	}
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);
	
	cin>>n;
	for(int i=1;i<=n;i++) scanf("%d",&p[i]);
	for(int i=1;i<=n;i++) {
		a[p[i]].x=1;
		b[500000-p[i]].x=1;
	}
	while((1<<bit)<=1000000) bit++;
	limit=1<<bit;
	for(int i=0;i<limit;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));
	fft(a,1); fft(b,1);
	for(int i=0;i<limit;i++) a[i]=a[i]*b[i];
	fft(a,-1);
	for(int i=500000;i<=1000000;i++) if((int)(a[i].x/limit+0.5)>0) c[i-500000]=1;
	for(int i=n;;i++) {
		bool flag=true;
		for(int j=i;j<N;j+=i) if(c[j]) { flag=false; break; }
		if(flag) {
			printf("%d\n",i);
			break;
		}
	}
	



	return 0;
}
/*

*/









  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值