天天看點

21牛客多校h(fft/ntt)

題目大意:找一個最小的m,使得n個數對m求模後無哈希沖突

思路:當 |ai - aj | ==k*m時存在哈希沖突,問題轉化為找最小m,使得任意兩個數的差不為m的倍數

大方向是從小到大枚舉m,看它的倍數是否存在即可,枚舉的複雜度是調和級數nlongn

判存在的話需要預處理

轉化成多項式乘法,若d存在則x^d的系數為1(不為0就行),反之為0

然後fft/ntt算下卷出來的系數,因為這裡是ai-aj,我們給後者加個5e5,避免出現負數

#include <cmath>
#include <cstring>
#include <algorithm>
#include <map>
#include <list>
#include <queue>
#include <vector>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <deque>
using namespace std;
namespace NTT
{
const long long g = 3;
const long long p = 998244353;
long long wn[35];
long long pow2(long long a, long long b)
{
    long long res = 1;
    while (b)
    {
        if (b & 1) res = res * a % p;
        a = a * a % p;
        b >>= 1;
    }
    return res;
}
void getwn()
{
    for (int i = 0; i < 25; i++) wn[i] = pow2(g, (p - 1) / (1LL << i));
}
void ntt(long long *a, int len, int f)
{
    long long i, j = 0, t, k, w, id;
    for (i = 1; i < len - 1; i++)
    {
        for (t = len; j ^= t >>= 1, ~j & t;);
        if (i < j) swap(a[i], a[j]);
    }
    for (i = 1, id = 1; i < len; i <<= 1, id++)
    {
        t = i << 1;
        for (j = 0; j < len; j += t)
        {
            for (k = 0, w = 1; k < i; k++, w = w * wn[id] % p)
            {
                long long x = a[j + k], y = w * a[j + k + i] % p;
                a[j + k] = (x + y) % p;
                a[j + k + i] = (x - y + p) % p;
            }
        }
    }
    if (f)
    {
        for (i = 1, j = len - 1; i < j; i++, j--) swap(a[i], a[j]);
        long long inv = pow2(len, p - 2);
        for (i = 0; i < len; i++) a[i] = a[i] * inv % p;
    }
}
void mul(long long *a, long long *b, int l1, int l2)
{
    int len, i;
    for (len = 1; len <= l1 + l2; len <<= 1);
    for (i = l1 + 1; i <= len; i++) a[i] = 0;
    for (i = l2 + 1; i <= len; i++) b[i] = 0;
    ntt(a, len, 0); ntt(b, len, 0);
    for (i = 0; i < len; i++) a[i] = a[i] * b[i] % p;
    ntt(a, len, 1);
}
};
typedef long long ll;
#define _for(i,a,b) for(int i=(a) ;i<=(b) ;i++)
#define _rep(i,a,b) for(int i=(a) ;i>=(b) ;i--)
#define scd(v) scanf("%d",&v)
#define scdd(a,b) scanf("%d %d",&a,&b)
#define endl "\n"
#define IOS ios::sync_with_stdio(false)
#define pb push_back
#define all(v) v.begin(),v.end()
#define int long long
#define odd(x) x&1
#define mst(v,a) memset(v,a,sizeof(v))
#define lson p<<1 ,l,mid
#define rson p<<1|1,mid+1,r
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define pii pair<double,double>
#define inf 0x7f7f7f7f
const int N=5e5+10;
//const int mod=998244353;
int n,m,k;
int ki_1[N*3],ki_2[N*3];
void solve();
signed main()
{
    //!!//
//    freopen("data.txt","r",stdin);
    //!!
    IOS;
    NTT::getwn();
    cin>>n;
    _for(i,1,n)
    {
        int x;cin>>x;
        ki_1[x]++;//ai系數
        ki_2[500000-x]++;//-aj系數
    }
    NTT::mul(ki_1,ki_2,5e5,5e5);//ntt加速求ki_1與系數ki_2的卷積
    bool ok=1;
    for(int i=1;;i++)//枚舉m
    {
        ok=1;
        for(int j=i ;j<=5e5 ;j+=i)//枚舉m的倍數
        {
            if( ki_1[j+500000] )
            {
                ok=0;
                break;
            }
        }
        if( ok )
        {
            cout<<i<<endl;
            return 0;
        }
    }
}
           

繼續閱讀