CodeForces 87 C.Interesting Game（博弈论+数论）

Description

两个人玩游戏，起初有一堆$n$个石子，每步一个人可以选取一堆将其分成$k$堆使得$k\ge 2$，且每堆石子的数量$a_1,...,a_k$满足$a_2-a_1=...=a_k-a_{k-1}=1$，如果某人无法完成该操作则失败，问先手第一次操作最少分成几堆可以必胜

Input

一个整数$n(1\le n\le 10^5)$

Output

若先手必胜则输出其第一步$k$值的最小值，否则输出$-1$

Sample Input

3

Sample Output

2

Solution

若$n$可以分成$k,...,k+j-1$这$j$堆，那么有$\frac{j(2k+j-1)}{2}=n$，也即$j(2k+j-1)=2n$，由于$k\ge 1$，故$j<2k+j-1$，也即$j$必然是$n$的因子且$j<\sqrt{n}$

因此对于每个石子堆数$i$直接枚举其因子$j$，判断是否存在合法的$k$，若存在则可以求出该种划分的$sg$值，显然后继状态不会超过$\sqrt{i}$个，直接根据这些后继状态的$sg$值即可求出当前状态的$sg$值，注意到对于一种划分方案$i=k+(k+1)+...+(k+j-1)$，其$sg$值为$Xor_{k\le l\le k+j-1}sg[l]=sum[k+j-1]\ Xor\ sum[k-1]$，维护$sg$值前缀和$sum$即可加速该步，时间复杂度$O(n\sqrt{n})$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100001;
int n,sg[maxn],f[maxn],sum[maxn];
int main()
{
    int ans=-1;
    scanf("%d",&n);
    sg[1]=sg[2]=0;
    for(int i=3;i<=n;i++)
    {
        for(int j=2;j*j<=2*i;j++)
            if(2*i%j==0)
            {
                int k=2*i/j-(j-1);
                if(k>0&&k%2==0)
                {
                    k/=2;
                    int temp=sum[k+j-1]^sum[k-1];
                    f[temp]=i;
                    if(i==n&&!temp&&ans==-1)ans=j;
                }
            }
        for(int j=0;;j++)
            if(f[j]!=i)
            {
                sg[i]=j;
                sum[i]=sum[i-1]^sg[i];
                break;
            }
    }
    printf("%d\n",ans);
    return 0;
}