POJ - 3376ファインディング回文manacher +トライ

問題の意味

文字列にN 2 2つのスプライシング、ステッチの後に尋ねた\(nは\ n倍\)回文文字列の数の文字列。

分析

すべての文字列の接頭辞と接尾辞の文字列を実行した辞書木、馬車にすべての正の文字列が辞書ツリーへのクエリで遊んで、位置を記録し、回文で、3通りの方法で2つの文字列回文文字列を構成します:

  • エンド一致しない、サフィックス文字列をプレイすることはパリンドロームです。

  • 最後のマッチング、それはサフィックス文字列が回文文字列です。

  • 最後にマッチする文字列は、n-演奏に等しいです。

現在の位置と正の数の接尾文字列回文文字列と正弦現在位置の数の辞書ツリーのメンテナンス。

コード

#include<cstring>
#include<cstdio>
#include<vector>
#include<iostream>
#include<algorithm>
#include<map>
#define fi first
#define se second
#define pb push_back
#define lson l,mid,p<<1
#define rson mid+1,r,p<<1|1
#define ll long long
using namespace std;
const int inf=1e9;
const int mod=1e9+7;
const int maxn=2e6+10;
int p[maxn*2],f[2][maxn],len[maxn],sum[2][maxn];
char s[maxn],t[maxn*2];
int son[maxn][26],tot;
ll ans;
void ins(int dl,int dr){
    int rt=0;
    for(int i=dl;i<=dr;i++){
        if(f[1][i]) sum[1][rt]++;
        if(!son[rt][s[i]-'a']) son[rt][s[i]-'a']=++tot;
        rt=son[rt][s[i]-'a'];
        if(i==dr) sum[0][rt]++;
    }
}
void qy(int dl,int dr){
    int rt=0;
    for(int i=dr;i>=dl;i--){
        if(f[0][i]) ans+=sum[0][rt];
        if(!son[rt][s[i]-'a']) return;
        rt=son[rt][s[i]-'a'];
        if(i==dl) ans+=sum[1][rt]+sum[0][rt];
    }
}
int mlc(int dl,int dr){
    int m=0,p0=1;p[1]=1;t[++m]='#';
    for(int i=dl;i<=dr;i++){
        t[++m]=s[i];
        t[++m]='#';
    }
    for(int i=2;i<=m;i++){
        int j=min(p[p0]+p0-i,p[2*p0-i]);
        if(i+j<p[p0]+p0){
            p[i]=j;
        }else{
            while(i-j>=1&&i+j<=m&&t[i-j]==t[i+j]) ++j;
            p[i]=j;p0=i;
        }
    }
    for(int i=dl;i<dr;i++){
        if(p[i-dl+2]-1==i-dl+1) f[0][i]=1;
        if(p[dr-2*dl+i+3]-1==dr-i) f[1][i+1]=1;
    }
}
int main(){
    //ios::sync_with_stdio(false);
    //freopen("in","r",stdin);
    int n;
    scanf("%d",&n);int l=1;
    for(int i=1,m;i<=n;i++){
        scanf("%d%s",&len[i],s+l);
        mlc(l,l+len[i]-1);
        ins(l,l+len[i]-1);
        l+=len[i];
    }l=1;
    for(int i=1;i<=n;i++){
        qy(l,l+len[i]-1);
        l+=len[i];
    }
    cout<<ans<<'\n';
    return 0;
}

おすすめ

転載: www.cnblogs.com/xyq0220/p/11511779.html