LOJ 6436 「PKUSC2018」神仙的游戏——思路+卷积

题目：https://loj.ac/problem/6436

看题解才会。

有长为 i 的 border ，就是有长为 n-i 的循环节。

考虑如果 x 位置上是 0 、 y 位置上是 1 ，那么长度是 | x-y | 的约数的循环节都不可行，因为在该循环节中， x 和 y 处在 “应该相等” 的地位。

最后一个部分分是暴力枚举 0 和 1 来预处理出一个 h[ i ] 表示长度是 i 的约数的循环节不可行。然后枚举循环节的长度 i ，再枚举 i 的倍数看看有没有 “不可行” 的。这样是 nlogn 。

考虑用卷积来优化求 h[ ] 。就是想找 “位置差一定” 的一个 0 和一个 1 ；令 \( F(x) = \sum [ s[i]=='0' ] x^i \) ，\( G(x) = \sum [ s[i]=='1' ] x^i \) ，翻转其中一个，做卷积即可。

感觉很卡常。没有 ' ? ' 的那个子任务，自己必须特判（用 kmp 做）才能不超时。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=5e5+,M=(<<)+,mod=;
int upt(int x){while(x>=mod)x-=mod;while(x<)x+=mod;return x;}
int pw(int x,int k)
{int ret=;while(k){if(k&)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=;}return ret;}

int n,f[M],g[M],len,r[M];char s[N];
int wn[M],wn2[M],nxt[N];
void ntt_init()
{
  for(int R=;R<=len;R<<=)
    {
      wn[R]=pw(,(mod-)/R);
      wn2[R]=pw(,(mod-)-(mod-)/R);
    }
}
void ntt(int *a,bool fx)
{
  for(int i=;i<len;i++)
    if(i<r[i])swap(a[i],a[r[i]]);
  for(int R=;R<=len;R<<=)
    {
      int Wn=(fx?wn2[R]:wn[R]);
      for(int i=,m=R>>;i<len;i+=R)
    for(int j=,w=;j<m;j++,w=(ll)w*Wn%mod)
      {
        int x=a[i+j],y=(ll)w*a[i+m+j]%mod;
        a[i+j]=upt(x+y); a[i+m+j]=upt(x-y);
      }
    }
  if(!fx)return; int inv=pw(len,mod-);
  for(int i=;i<len;i++)a[i]=(ll)a[i]*inv%mod;
}
void kmp()
{
  for(int i=n;i;i--)s[i]=s[i-];
  for(int i=;i<=n;i++)
    {
      int cr=nxt[i-];
      while(cr&&s[cr+]!=s[i])cr=nxt[cr];
      if(s[cr+]==s[i])nxt[i]=cr+;
      else nxt[i]=;
    }
  ll ans=(ll)n*n;
  int cr=nxt[n];
  while(cr)
    {
      ans^=(ll)cr*cr; cr=nxt[cr];
    }
  printf("%lld\n",ans);
}
int main()
{
  scanf("%s",s); n=strlen(s); bool chk=;
  for(int i=;i<n;i++)
    {
      f[i]=(s[i]==''); g[n--i]=(s[i]=='');
      if(s[i]=='?')chk=;
    }
  if(!chk){ kmp();return ;}
  for(len=;len<n<<;len<<=);
  for(int i=,j=len>>;i<len;i++)
    r[i]=(r[i>>]>>)+((i&)?j:);
  ntt_init();
  ntt(f,); ntt(g,);
  for(int i=;i<len;i++)f[i]=(ll)f[i]*g[i]%mod;
  ntt(f,);
  for(int i=;i<n;i++)g[i]=((f[n--i]||f[n-+i])?:);
  ll ans=(ll)n*n;
  for(int i=;i<n;i++)
    {
      int x=n-i; bool fg=;
      for(int j=x;j<n;j+=x)
    if(g[j]){fg=;break;}
      if(!fg)ans^=(ll)i*i;
    }
  printf("%lld\n",ans);
  return ;
}