Cyclical Quest CodeForces - 235C (后缀自动机 SAM)

题目大意:
题目大意:给一个字符串S,再给一个字符串T,设T的长度为len,问T的循环串在S中出现的次数,这里循环串的定义是:对于一个长度为len的字符串,我们把它首尾相接,然后从任意位置开始走len步所得到的串我们叫做T的循环串。如abaa的循环串有 abaa,baaa,aaab,aaba。(注意如果重复只算一次。比如aaa的循环串只有一个aaa)

参考http://blog.csdn.net/dyx404514/article/details/8807278

感悟:
这个地方的基数排序原来是这样的。。

#include<bits/stdc++.h>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define sf scanf
#define pf printf
#define mem(a,b) memset(a,b,sizeof(a));
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define MP make_pair
#define ULL unsigned long long
#define LL   long long
#define inf 0x3f3f3f3f
#define md ((ll+rr)>>1)
#define ls (i<<1)
#define rs (ls|1)
#define eps 1e-5
#define N 2000010
#define ree freopen("in.txt","r",stdin);
#define bug pf("----------------");

struct state{
    state*par,*son[26];
    int val,vis,num;
    void init(){
        par=NULL;
        mem(son,0);
        val=0;
        vis=num=0;
    }
}*root,*last,que[N];

int tot;
void  extend(int w){
    state*p=last;
    que[tot].init();
    state*np=&que[tot++];
    np->val=p->val+1;
    while(p&&p->son[w]==0)
        p->son[w]=np,p=p->par;
    if(p==0)
        np->par=root;
    else{
        state*q=p->son[w];
        if(q->val==p->val+1)
            np->par=q;
        else{
            state*nq=&que[tot++];
            *nq=*q;
            nq->val=p->val+1;
            q->par=np->par=nq;
            while(p&&p->son[w]==q){
                p->son[w]=nq,p=p->par;
            }
        }
    }
    last=np;
}
char s[N];
int rk[N],cnt[N];
int main(){
    //ree
    sf("%s",s);
    tot=0;
    que[0].init();
    root=last=&que[tot++];
    int n=strlen(s);
    for(int i=0;i<n;++i)
        extend(s[i]-'a');
    state* cur=root;
    for(int i=0;i<n;++i)cur=cur->son[s[i]-'a'],cur->num++;
    for(int i=0;i<tot;++i)++cnt[que[i].val];
    for(int i=1;i<=n;++i)cnt[i]+=cnt[i-1];
    for(int i=0;i<tot;++i) rk[--cnt[que[i].val]]=i;
    for(int i=tot-1;i>=0;--i){
        int v=rk[i];
        if(que[v].par)que[v].par->num+=que[v].num;
    }
    int t;
    sf("%d",&t);
    while(t--){
        sf("%s",s);
        int len=strlen(s);
        for(int i=len;i<len*2;++i)s[i]=s[i-len];
        cur=root;
        int l=0;
        int ans=0;
        for(int i=0;i<len*2;++i){
            int c=s[i]-'a';
            if(cur->son[c]){
                ++l;
                cur=cur->son[c];
            }
            else{
                while(cur&&cur->son[c]==0)cur=cur->par;
                if(cur==NULL)cur=root,l=0;
                else l=cur->val+1,cur=cur->son[c];
            }
            if(l>=len){
                while( cur->par && cur->par->val >=len)cur=cur->par;
                if( cur->vis !=t+1)cur->vis=t+1,ans+=cur->num;
            }
        }
        pf("%d\n",ans);
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值