Wannafly挑战赛23 T2游戏 SG函数

哎，被卡科技了，想了三个小时，最后还是大佬给我说是\(SG\)函数。
\(SG\)函数，用起来很简单，证明呢？（不可能的，这辈子都是不可能的）

\(SG\)定理

游戏的\(SG\)函数就是各个子游戏的\(SG\)函数的\(Nim-sum\)（就是异或和），比如多堆石子的\(SG\)函数就是所有单堆石子\(SG\)函数的异或和。

\(SG\)函数

首先定义\(mex(T)\)为\(T\)中未出现的自然数中最小的数，其中\(T \subset N\)，如\(mex(0,2,3)=1\)，\(mex(4,7)=0\)。那么\(SG(x)=mex(S)\)，\(S\)为\(x\)的后继状态的\(SG\)函数值集合。
然后，对于某一个状态\(x\)，若\(SG(x)=0\)，则先手必败，否则先手必胜。
对于这一道题，因为只能拿约数个，也就是至少拿一个，所以我们可以先从小到大预处理一下\(SG\)函数。然后因为我们是先手，要给对手制造一个必败状态，所以只要枚举一下第一步的所有策略，然后判断一下剩下的那些石子的\(SG(x)\)函数的异或和是否为\(0\)就行了。
期望复杂度\(O(n^{\frac{3}{2}})\)。

#include <bits/stdc++.h>
 
using namespace std;
 
const int N = 100000;
 
int n, a[N+5], sum, p, ans, SG[N+5], vis[N+5];
 
void calc_SG() { //预处理SG函数
    SG[0] = 0, SG[1] = 1; //初始化
    for(int i = 2; i <= N+1; i++) {
        int cnt = 0, t = 0x3f3f3f3f;
        for(int j = 1; j*j <= i; j++)
            if(i%j == 0) {
                vis[++cnt] = SG[i-j];
                if(j*j != i) vis[++cnt] = SG[i-i/j];
            }
        sort(vis+1, vis+cnt+1);
        if(vis[1] >= 1) { //计算mex(T)
            SG[i] = 0;
            continue;
        }
        for(int j = 1; j <= cnt-1; j++)
            if(vis[j+1]-vis[j] > 1) t = min(t, vis[j]+1);
        t = min(t, vis[cnt]+1);
        SG[i] = t;
    }
}
 
int main() {
    ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0); //读入输出加速
    cin >> n;
    calc_SG(); //预处理SG函数
    for(int i = 1; i <= n; i++) cin >> a[i], sum ^= SG[a[i]];
    for(int i = 1; i <= n; i++) {
        p = sum^SG[a[i]];
        for(int j = 1; j*j <= a[i]; j++) {
            if(a[i]%j) continue;
            if((p^SG[a[i]-j]) == 0) ans++; //判断剩下的Nim-sum是否为0
            if(j*j != a[i] && (p^SG[a[i]-a[i]/j]) == 0) ans++;
        }
    }
    cout << ans << endl;
    return 0;
}