HDU多校第二場 - hdoj6599 I Love Palindrome String

題目大意:給一個長度n的字符串S,問長度分別爲1 - n 的迴文子串個數,且每個迴文子串S(l,r) 需滿足 S(l,(l+r)/2)也是迴文串。

       迴文自動機處理出不同迴文子串的個數,然後用馬拉車判斷每個迴文串是否滿足要求即可。。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
char s[300010],ss[700000];
int len[700000];
int f[300010],T[300010][30],L[300010],cnt,vis[300010];
long long sum[300010],ans[300010];
void insert(int n){
    int i,v,k,x=0;
    f[0]=1;
    L[1]=-1;
    s[0]='#';
    cnt=1;
    for(i=1;i<=n;i++){
        v=s[i]-'a';
        while(s[i]!=s[i-L[x]-1]) x=f[x];
        if(T[x][v]==0){
             L[++cnt]=L[x]+2;
             k=f[x];
             while(s[i]!=s[i-L[k]-1]) k=f[k];
             f[cnt]=T[k][v];
             T[x][v]=cnt;
             vis[cnt]=i;
        }
        x=T[x][v];
        sum[x]++;
    }
}

int main(){
    int i,n;
    while(scanf("%s",s+1)!=EOF){
         n=strlen(s+1);
         for(i=0;i<=2*n+100;i++) len[i]=0;
         for(i=0;i<=n+5;i++){
              sum[i]=ans[i]=L[i]=vis[i]=f[i]=0;
              memset(T[i],0,sizeof(T[i]));
         }
         ss[0]='*';
         for(i=1;i<=n+1;i++){
              ss[i*2]=s[i];
              ss[i*2-1]='#';
         }
         int mx=0,p=0,k;
         for(i=1;i<=2*n;i++){
             if(mx>i) len[i]=min(mx-i,len[2*p-i]);
             else len[i]=1;
             while(ss[i-len[i]]==ss[i+len[i]]) len[i]++;
             if(i+len[i]>mx){
                 mx=i+len[i];
                 p=i;
             }
         }
         insert(n);
         for(i=cnt;i;i--) sum[f[i]]+=sum[i];
         int l,r,mid,ll;
         for(i=2;i<=cnt;i++){
              l=vis[i]-L[i]+1;
              r=vis[i]-L[i]/2;
              l*=2;
              r*=2;
              mid=(l+r)/2;       //找回文中心
              ll=len[mid]-1;
              if(ll*2>=L[i]) ans[L[i]]+=sum[i];
         }
         for(i=1;i<n;i++) printf("%lld ",ans[i]);
         printf("%lld\n",ans[n]);
    }
    return 0;
}

 

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章