BZOJ3103 : Palindromic Equivalence

Claris發表於2015-06-22

用Manacher可以推出O(n)對相等和不等關係。

將相等的用並查集維護,不等的連邊。

然後從1到n,如果該等價類還沒被考慮過,則ans*=26-與它不等的考慮過的等價類個數。

 

#include<cstdio>
#include<cstring>
#define N 1000010
int n,m,i,r,p,f[N<<1],g[N],vis[N],ans=1;char a[N],s[N<<1],col[N];
struct E{int v;E*nxt;}*j,*h[N],pool[N*2],*cur=pool;
inline int min(int a,int b){return a<b?a:b;}
int F(int x){return g[x]==x?x:g[x]=F(g[x]);}
inline void merge(int x,int y){
  if(x&1)return;
  if(F(x>>=1)==F(y>>=1))return;
  g[g[x]]=g[y];
}
inline void add(int x,int y){
  if(!x||!y||x&1)return;
  j=cur++;j->v=y>>1;j->nxt=h[x>>=1];h[x]=j;
}
int main(){
  scanf("%s",a+1),n=std::strlen(a+1);
  for(i=1;i<=n;i++)g[i]=i;
  for(i=1;i<=n;i++)s[i<<1]=a[i],s[i<<1|1]='#';s[0]='$',s[1]='#',s[m=(n+1)<<1]='@';
  for(f[1]=1,i=2;i<m;i++){
    for(f[i]=r>i?min(r-i,f[p*2-i]):1;s[i-f[i]]==s[i+f[i]];f[i]++)merge(i+f[i],i-f[i]);
    add(i+f[i],i-f[i]);
    if(i+f[i]>r)r=i+f[i],p=i;
  }
  for(i=1;i<=n;i++)if(!col[F(i)]){
    for(p=26,j=h[i];j;j=j->nxt)if(col[F(j->v)])if(vis[g[j->v]]<i)vis[g[j->v]]=i,p--;
    col[g[i]]=1,ans=1LL*ans*p%1000000007;
  }
  return printf("%d",ans),0;
}