码迷,mamicode.com
首页 > 其他好文 > 详细

循环移位(后缀自动机)

时间:2019-08-20 22:30:24      阅读:115      评论:0      收藏:0      [点我收藏+]

标签:ret   width   nbsp   oid   scan   char   clu   描述   ace   

题目描述:

给定一个字符串 s 。现在问你有多少个本质不同的 s 的子串 t=ttt(m>0使得将 t 循环左移一位后变成的 t=ttt1 也是 s 的一个子串。

题解:

技术图片

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=300010;
char s[N];
int n,lst=1,tot=1;
int len[N*2],ch[N*2][26],fa[N*2],pos[N*2];
//pos表示在原串中的位置
int sum[N][26];
ll ans;
void add(int c){
    int p=lst,np=lst=++tot;
    pos[np]=len[np]=len[p]+1;
    for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
    if(!p) fa[np]=1;
    else{
        int q=ch[p][c];
        if(len[q]==len[p]+1) fa[np]=q;
        else{
            int nq=++tot;
            memcpy(ch[nq],ch[q],sizeof(ch[q]));
            fa[nq]=fa[q];
            len[nq]=len[p]+1;
            pos[nq]=pos[q];
            fa[q]=fa[np]=nq;
            for(;p&&ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
        }
    }
    return;
}
int main(){
    scanf("%s",s+1);
    n=strlen(s+1);
    for(int i=1;i<=n;i++)
        add(s[i]-a);
    for(int i=1;i<=n;i++)
        for(int j=0;j<26;j++)
            sum[i][j]=sum[i-1][j]+(s[i]-a==j);
    for(int i=2;i<=tot;i++){
        if(ch[fa[i]][s[pos[i]-len[fa[i]]]-a])
            ans++;
        for(int j=0;j<26;j++)
            if(ch[i][j])
                ans+=sum[pos[i]-len[fa[i]]-1][j]-sum[pos[i]-len[i]][j];
    }
    printf("%lld",ans);
    return 0;
}

好了我知道这个不好懂,所以我们再配张图:

技术图片

 

循环移位(后缀自动机)

标签:ret   width   nbsp   oid   scan   char   clu   描述   ace   

原文地址:https://www.cnblogs.com/HarryPotter-fan/p/11385786.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!