【uva11855-求长度为1到n的相同子串出现的次数】sam

题意：求长度为1到n的相同子串出现的次数，输到小于2为止。

题解：

用sam做。

建机，算right集合，然后用r[i]更新长度为step[i]的子串出现次数，然后ans[i]=maxx(ans[i],ans[i+1])（长度更长的出现次数一定小于等于长度更短的。）

 

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;

const int N=2*100010;
int sl,cl,tot,last,step[N],pre[N],son[N][30],in[N],r[N],c[N],ans[N];
bool vis[N];
char s[N];
queue<int> Q;

int maxx(int x,int y){return x>y ? x:y;}

int add_node(int x)
{
    step[++tot]=x;
    return tot;
}

void clear()
{
    memset(step,0,sizeof(step));
    memset(pre,0,sizeof(pre));
    memset(son,0,sizeof(son));
    memset(in,0,sizeof(in));
    memset(r,0,sizeof(r));
    tot=0;add_node(0);last=1;
}

void extend(int ch)
{
    int p=last,np=add_node(step[p]+1);
    while(p && !son[p][ch])
    {
        son[p][ch]=np;
        in[np]++;
        p=pre[p];
    }
    if(!p) pre[np]=1;
    else
    {
        int q=son[p][ch];
        if(step[q]==step[p]+1) pre[np]=q;
        else
        {
            int nq=add_node(step[p]+1);
            for(int i=1;i<=26;i++) 
                if(son[q][i]) son[nq][i]=son[q][i],in[son[q][i]]++;
            pre[nq]=pre[q];
            pre[np]=pre[q]=nq;
            while(p && son[p][ch]==q) in[q]--,in[nq]++,son[p][ch]=nq,p=pre[p];
        }
    }
    last=np;
}

void get_tp()
{
    while(!Q.empty()) Q.pop();
    memset(vis,0,sizeof(vis));
    Q.push(1);vis[1]=1;cl=0;
    while(!Q.empty()) 
    {
        int x=Q.front();c[++cl]=x;vis[x]=0;Q.pop();
        for(int i=1;i<=26;i++)
        {
            int y=son[x][i];
            if(!y) continue;
            in[y]--;
            if(!in[y] && !vis[y]) vis[y]=1,Q.push(y);
        }
    }
}

void get_right()
{
    int x=1,ch;
    for(int i=1;i<=sl;i++)
    {
        ch=s[i]-'A'+1;
        x=son[x][ch];
        r[x]++;
    }
    for(int i=cl;i>=1;i--) r[pre[c[i]]]+=r[c[i]];
}

void output(int x)
{
    printf("x = %d  pre = %d  r = %d\n",x,pre[x],r[x]);
    for(int i=1;i<=26;i++)
    {
        if(son[x][i]) printf("son %d = %d\n",i,son[x][i]);
    }
    printf("\n");
}

int main()
{
    freopen("a.in","r",stdin);
    // freopen("a.out","w",stdout);
    int cas=0;
    while(1)
    {
        gets(s+1);
        sl=strlen(s+1);
        int l=0;
        for(int i=1;i<=sl;i++) 
            if(s[i]>='A' && s[i]<='Z') s[++l]=s[i];
        sl=l;
        if(sl==0) return 0;

        clear();
        for(int i=1;i<=sl;i++) extend(s[i]-'A'+1);
        get_tp();
        get_right();
        memset(ans,0,sizeof(ans));
        for(int i=1;i<=tot;i++) ans[step[i]]=maxx(ans[step[i]],r[i]);

        for(int i=1;i<sl;i++) ans[i]=maxx(ans[i],ans[i+1]);
        
        for(int i=1;i<=sl;i++) 
        {
            if(ans[i]<2) break;
            printf("%d\n",ans[i]);
        }
        printf("\n");
    }
    return 0;
}