spoj 694,705 后缀数组，后缀自动机

      题意很简单，给一个字符串，求这个字符串有多少个不同的子串。这两题其实是一道，只不过694的范围是1000,705的范围是50000。写后缀数组的话，先求出sa,height这两个数组，因为每个后缀一定是一个独一无二的子串，所以可以直接把len先给ans，接下来就是找非后缀的子串，我的做法是直接循环sa，因为相邻的后缀已经是最相近的了，我要添加的就只是后缀i中从height[i]+1位开始到倒数第二位这些子串（height[i]位的情况上一层已经添加过，到最后一位的情况算到之前的len种情况里了）。循环完直接输出答案即可。代码见下。

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <algorithm>
#include <stack>
#include <queue>
#include <map>
#include <string>
#include <cstring>
#include <string>
using namespace std;
typedef long long ll;
const int maxn=80000+40;
int s[maxn],rs[maxn];
int sa[maxn],t[maxn],t2[maxn],c[maxn];
int n,m,k;
char s1[maxn],s2[maxn];
int rank[maxn],height[maxn];
int l1,l2;

void getheight(int n)
{
    int i,j,k=0;
    for (i=0; i<=n; i++) rank[sa[i]]=i;
    for (i=0; i<n; i++)
    {
        if (k) k--;
        int j=sa[rank[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rank[i]]=k;
    }
}

void build_ss(int m,int n)
{
    n++;
    int i,*x=t,*y=t2;
    for (int i=0; i<m; i++) c[i]=0;
    for (int i=0; i<n; i++) c[x[i]=s[i]]++;
    for (int i=1; i<m; i++) c[i]+=c[i-1];
    for (int i=n-1; i>=0; i--)
      sa[--c[x[i]]]=i;
    for (int k=1; k<=n; k<<=1)
    {
        int p=0;
        for (i=n-k; i<n; i++) y[p++]=i;
        for (i=0; i<n; i++) if (sa[i]>=k) y[p++]=sa[i]-k;

        for (i=0; i<m; i++) c[i]=0;
        for (i=0; i<n; i++) c[x[y[i]]]++;
        for (i=1; i<m; i++) c[i]+=c[i-1];
        for (i=n-1; i>=0; i--) sa[--c[x[y[i]]]] = y[i];
        swap(x,y);
        p=1;
        x[sa[0]]=0;
        for (i=1; i<n; i++)
        x[sa[i]]=(y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k])? p-1 : p++;
        if (p>=n) break;
        m=p;
    }
}
int tt,ans;
int main()
{
//    freopen("in.txt","r",stdin);
    scanf("%d",&tt);
    while(tt--)
    {

        scanf("%s",s1);
        int l=strlen(s1);
        n=l;
        m=0;
        for (int i=0; i<l; i++)
        {
            s[i]=s1[i];
            m=max(m,s[i]);
        }
        s[l]=0;
        build_ss(m+1,n);
        getheight(n);
        ans=n;
        int tp;
        for (int i=1; i<=n; i++)
        {
            tp=n-sa[i];
            ans+=(tp-1-height[i]);
        }
        printf("%d\n",ans);
    }
    return 0;
}

      另外，我感觉这题后缀自动机应该也能写，我写的时候是先建好了后缀自动机，从每一个节点开始，走到任何一个节点都是一个独一无二的子串，所以只要统计一下就行，1000的毫无压力的过掉了，但50000的TLE到死..DFS统计TLE，换成从起点一层一层的递推还是TLE..后来想了想这么统计复杂度严格来说应该是O（边数）而不是O（n），可能是这T掉了吧。有大神看到了还求指导这题后缀自动机的做法。代码还是贴一下吧。

//只能过694。705TLE

#include <vector>
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <memory.h>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <string>
#include <cstring>
using namespace std;
typedef long long ll;
typedef vector<int> vi;
const int maxn=100100;
const int S=256;
int tot,len;
int n,m;
int wtop[maxn];
int c[maxn];
int k;
int ans;
struct node
{
    node *par,*go[S];
    int val,sk;
    int id;
}*tail,*root,que[maxn],*top[maxn];
char str[maxn],s[maxn],ss[maxn];

inline int idx(char s)
{
//  if (s>='A' && s<='Z') return s-'A';
//  if (s>='a' && s<='z') return s-'a'+26;
//  return 52+s-'0';
    return (int)s;
}
inline void init(int m)
{
    for (int i=0; i<=m; i++)
    {
        memset(que[i].go,0,sizeof que[i].go);
        que[i].id=que[i].val=que[i].sk=0;
    }
    tot=0;
    len=1;
    root=tail=&que[tot++];
    root->id=0;
    ans=0;
}
void add(int c,int l)
{
    node* p=tail;
    node* np=&que[tot++];
    np->val=l;
    np->id=tot-1;
    while(p && p->go[c]==NULL) p->go[c]=np,p=p->par;
    if (p==NULL) np->par=root;
    else
    {
        node *q=p->go[c];
        if (p->val+1==q->val) np->par=q;
        else
        {
            node *nq=&que[tot++];
            *nq=*q;
            nq->id=tot-1;
            nq->val=p->val+1;
            np->par=q->par=nq;
            while(p && p->go[c]==q) p->go[c]=nq,p=p->par;
        }
    }
    tail=np;
}
void debug_suff()
{
    for (int i=0; i<tot; i++)
    {
        for (int c=0; c<S; c++)
        if (que[i].go[c])
        {
            cout<<que[i].id<<" "<<que[i].go[c]->id<<endl;
        }
    }
}
void debug_pre()
{
    for (int i=1; i<tot; i++)
    {
        cout<<que[i].id<<" "<<que[i].par->id<<endl;
    }
}
inline void TopS()
{
    memset(c,0,sizeof c);
    for (int i=0; i<tot; i++)
    c[que[i].val]++;
    for (int i=1; i<len; i++)
    c[i]+=c[i-1];
    for (int i=0; i<tot; i++)
    top[--c[que[i].val]]=&que[i],wtop[c[que[i].val]]=i;
}

int tt;
int main()
{
//    freopen("in.txt","r",stdin);
    scanf("%d",&tt);
    while(tt--)
    {

        scanf("%s",s);
        int l=strlen(s);
        init(l<<1);
        for (int i=0; i<l; i++)
        {
            add(idx(s[i]),len++);
        }

//        debug_suff();
//        puts(" ");
//        debug_pre();
        node *rt=root;
        rt->sk=1;
        TopS();
        for (int i=0; i<tot; i++)
        {
            rt=top[i];
            ans+=rt->sk;
            for(int c=0; c<S; c++)
            {
                if (rt->go[c])
                {
                    rt->go[c]->sk+=rt->sk;
                }
            }
        }
        printf("%d\n",ans-1);
    }
    return 0;
}