bzoj 5372: [Pkusc2018]神仙的游戏

Description

小D和小H是两位神仙。他们经常在一起玩神仙才会玩的一些游戏，比如“口算一个4位数是不是完全平方数”。

今天他们发现了一种新的游戏：首先称s长度为len的前缀成为border当且仅当

s[1…len]=s[|s|-len+1…|s|]。

给出一个由01?组成的字符串s,将s中的问号用变成01替换，对每个len口算是否存在替换问号的方案使得s长度为len的前缀成为border，

把这个结果记做f(len)∈{0,1}。f(len)=1如果s长度为len的前缀能够成为border，否则f(len)=0

由于小D和小H是神仙，所以他们计算的s的长度很长，因此把计算的结果一一比对会花费很长的时间。为了方便比对，他们规定了一个校验值：

(f(1)12)xor(f(2)22)xor(f(3)32)xor…xor(f(n)n2)

来校验他们的答案是否相同。xor表示按位异或。

但是不巧，在某一次游戏中，他们口算出的校验值并不一样，他们希望你帮助他们来计算一个正确的校验值。

当然，他们不强迫你口算，可以编程解决。

Solution

首先假设 $border=len$ , 那么相当于是把串 $s[1...len]$ 向右平移 $n-len$ , 然后这个串和平移后的相等.

也就是说存在一个长度为 $x$ 的循环节 , 那么就存在 $border=n-x$ , 所以判断是否存在循环节就行了.

对于一对 $i,j$ , $s[i]!=s[j]!='?'$ , 那么就不存在长度为 $|i-j|$ 的循环节 , 自然也不存在 $|i-j|$ 约数的循环节.

所以我们把不合法的判掉 , 剩下的就是循环节了 , 距离为 $i-j$ 的所有字符对一起考虑 , 只需要把 $j$ 变成 $n-j$ 做一边 $FFT$ ,通过位置 $i-j+n$ 就可以判断是否存在长度为 $i-j$ 的循环节了.

#include<bits/stdc++.h>

using namespace std;

template<class T>void gi(T &x){

	int f;char c;

	for(f=1,c=getchar();c<'0'||c>'9';c=getchar())if(c=='-')f=-1;

	for(x=0;c<='9'&&c>='0';c=getchar())x=x*10+(c&15);x*=f;

}

typedef complex<double> dob;

const int N=2e6+10;const double pi=acos(-1.0);

int n,m,R[N],L=0,d[N];char s[N];

inline void FFT(dob *A,int o){

	for(int i=0;i<n;i++)if(i<R[i])swap(A[i],A[R[i]]);

	for(int i=1;i<n;i<<=1){

		dob wn(cos(pi/i),sin(pi*o/i)),x,y;

		for(int j=0;j<n;j+=i<<1){

			dob w(1,0);

			for(int k=0;k<i;k++,w=w*wn){

				x=A[j+k],y=w*A[j+k+i];

				A[j+k]=x+y,A[j+k+i]=x-y;

			}

		}

	}

}

dob A[N],B[N];

int main(){

  freopen("pp.in","r",stdin);

  freopen("pp.out","w",stdout);

  scanf("%s",s+1),m=strlen(s+1);

  for(n=1;n<=(m<<1);n<<=1)L++;

  for(int i=0;i<n;i++)R[i]=(R[i>>1]>>1)|((i&1)<<(L-1));

  for(int i=1;i<=m;i++)A[m-i]=(s[i]=='1'),B[i]=(s[i]=='0');

  FFT(A,1),FFT(B,1);

  for(int i=0;i<=n;i++)A[i]*=B[i];

  FFT(A,-1);

  for(int i=m+1;i<n;i++)d[i-m]=(int)(A[i].real()/n+0.5)|(int)(A[2*m-i].real()/n+0.5);

  for(int i=m;i>=1;i--)

	  if(!d[i])for(int j=i;j<=m;j+=i)if(d[j]){d[i]=1;break;}

  long long ans=0;

  for(int i=1;i<=m;i++)if(!d[m-i])ans^=1ll*i*i;

  cout<<ans;

  return 0;

}