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

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

题目链接：牛客多校2 H题 - Hash Function

题意

给定一个序列$a_n$要求寻找一个最小的模数使得对序列$a_n$的所有数对该数取模后没有冲突（即取模后不存在两个相同的数）

分析

设最小模数为$p$，要使$a_n$对$p$取模后不存在冲突，则不存在$a_j\mathrm{m}\mathrm{o}\mathrm{d}p\equiv a_j\mathrm{m}\mathrm{o}\mathrm{d}p$，即$|a_i-a_j|\mathrm{m}\mathrm{o}\mathrm{d}p\equiv 0$

由此题意转化为求最小的$p$，使得$p$和它的倍数不为任意两个$a_i$与$a_j$差值

也就是说，我们要在题目规定的时间范围内求得所有$|a_i-a_j|$，由于$N=5e5$，所以$O(N^2)$的做法会爆掉，(然而出题人在比赛时居然没把另一种暴力的做法卡掉，被很多人水了过去，结果赛后加强了数据，把我们这些拉题自训的给卡了orz）

所以应该用FFT/NTT来处理，把复杂度降到$O(N\log N)$

我们可以用类似生成函数的思想来构造一个多项式，设 b i ∈ { 0 , 1 } b_i \in \{0,1\} bi​∈{0,1}，当且仅当$i$在$a$中出现过时,$b_i=1$，令：

$B(x)={\sum }_{i=0}^n{b}_i{x}^i$

$C(x)={\sum }_{i=0}^n{c}_i{x}^i={\sum }_{i=0}^n{b}_{n-i}{x}^i$

$D(x)=B(x)C(x)={\sum }_{i=0}^{2n}{d}_i{x}^i$

其中

$d_i={\sum }_{j=0}^i{b}_i{c}_{i-j}={\sum }_{j=0}^i{b}_j{b}_{n-i+j}$

则$d_i>0$的含义为，至少存在一对$j$和$n-i+j$同时在$a$中出现过，即存在$u,v$使得$|a_u-a_v|=i-n$，那么我们用FFT求出$D(x)=B(x)C(x)$后，枚举$[n,2n]$之间的$d_i$的值，就可以求出$[0,n]$内每个值是否为$a$中某两个数的差值，这样就可以在$O(N\log N)$的时间内求解

参考：【题解】2021牛客暑期多校第一场 H.Hack function

​ 倾心讲解 FFT 多项式与快速傅里叶变换 && 迭代模板带注释

#pragma GCC optimize(3, "Ofast", "inline")
#include <bits/stdc++.h>

const int N = 500005, FFTN = N << 2;
const double PI = acos(-1);

struct Complex {
    double x, y;
    Complex() {}
    Complex(double _x, double _y) : x(_x), y(_y) {}
    Complex operator + (const Complex &r) const {
        return Complex(x + r.x, y + r.y);
    }
    Complex operator - (const Complex &r) const {
        return Complex(x - r.x, y - r.y);
    }
    Complex operator * (const Complex &r) const {
        return Complex(x*r.x - y*r.y, x*r.y+y*r.x);
    }
} a[FFTN], b[FFTN];

int n;

int fft_n = 1, fft_m, R[FFTN];

void init() {
    int len = N << 1; //**//
    while(fft_n <= len) fft_n <<= 1, ++fft_m;
    for(int i = 0; i < fft_n; ++i)
        R[i] = (R[i>>1]>>1) | ((i&1) << (fft_m-1));
}

void FFT(Complex *A, int f) {
    for(int i = 0; i < fft_n; ++i) {
        if(i < R[i]) std::swap(A[i], A[R[i]]);
    }
    for(int i = 1; i < fft_n; i <<= 1) {
        Complex wn(cos(PI/i), f*sin(PI/i));
        for(int j = 0, r = (i << 1); j < fft_n; j += r) {
            Complex w(1, 0);
            for(int k = 0; k < i; ++k, w = w * wn) {
                Complex x = A[j+k], y = w * A[i+j+k];
                A[j+k] = x + y, A[i+j+k] = x - y;
            }
        }
    }
    if(f == -1) {
        for(int i = 0; i < fft_n; ++i)
            A[i].x /= fft_n;
    }
}

bool vis[N<<1];

int main() {
    scanf("%d", &n);
    for(int i = 0, val; i < n; ++i) {
        scanf("%d", &val);
        a[val].x = 1;
        b[N-val].x = 1;
    }
    init();
    FFT(a, 1); FFT(b, 1);
    for(int i = 0; i < fft_n; ++i)
        a[i] = a[i] * b[i];
    FFT(a, -1);
    for(int i = N; i <= (N << 1); ++i)
        vis[i-N] = int(a[i].x +  0.5);
    for(int i = n; i <= N; ++i) {
        bool flag = false;
        for(int j = i; j <= N; j += i)
            flag |= vis[j];
        if(!flag) {
            printf("%d", i);
            break;
        }
    }
    return 0;
}

一些没什么用的注意事项

• 抄完 写完FFT板子后在本地运行发现居然跑出了2000ms+，慢的要死，放在队友电脑上跑出来也是一样的慢，我寻思题目要求2s这能过？于是把代码放牛客上交了一发300ms+过了，相距将近10倍，差点以为见鬼了。后来在各种试错+探索下发现本地加上O3优化（就是代码打头那句）后也能跑出300ms+，搞半天是牛客自动开启了O3优化，良心OJ了属于是