【hdu4436/LA6387-str2int】sam处理不同子串

题意：给出n个数字，数字很长，用字符串读入，长度总和为10^5。求这n个字符串的所有子串（不重复）的和取模2012 。

例如字符串101，和就是1+10+101=112。

题解：

就是求不同的子串连成一个数。
sam的拓扑序真的很有用！按拓扑序可以保证能转移到当前x的节点都在之前被更新了。
每个节点x维护cnt表示root到x的方案数，sum表示以x为结尾的子串和。
找拓扑序有两种方法：1.拓扑排序一样bfs(O(n)) 2.按照step[x]排序（若x有个孩子是y，step[x]<=step[y](O(nlogn))
按照拓扑序for一遍，对于x，它的孩子y,cnt[y]+=cnt[x],sum[y]+=sum[x]*10+cnt[x]*(y所代表的数字)。
不能有前缀0----->根节点不走0孩子。
节点数开了10^6才AC。tell me why?!

打了两种求拓扑序的方法。

贴一下代码。

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

 const int N=*,Mod=;
 char s[N];
 int n,sl,cl,ans,tot,last;
 int son[N][],pre[N],step[N],sum[N],in[N],cnt[N],c[N];
 bool vis[N];
 queue<int> Q;

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

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

 void find_tp_1()
 {
     for(int i=;i<=tot;i++)
     {
         if(in[i]==)
         {
             for(int j=;j<=;j++)
             {
                 int y=son[i][j];
                 if(!y) continue;
                 in[y]--;
             }
         }
     }
     while(!Q.empty()) Q.pop();
     Q.push();vis[]=;cl=;
     while(!Q.empty())
     {
         int x=Q.front();vis[x]=;c[++cl]=x;Q.pop();
         for(int i=;i<=;i++)
         {
             int y=son[x][i];
             if(!y) continue;
             in[y]--;
             if(!in[y] && !vis[y]) vis[y]=,Q.push(y);
         }
     }
 }

 bool cmp(int x,int y){return step[x]<step[y];}
 void find_tp_2()
 {
     cl=;
     for(int i=;i<=tot;i++)
     {
         if(in[i] || i==) c[++cl]=i;
     }
     sort(c+,c++cl,cmp);
 }

 void solve()
 {
     cnt[]=;
     for(int i=;i<=cl;i++)
     {
         int x=c[i];
         for(int j=;j<=;j++)
         {
             int y=son[x][j];
             if(!y || (x== && j==)) continue;
             cnt[y]=(cnt[y]+cnt[x])%Mod;
             sum[y]=(sum[y]+(sum[x]*)%Mod+(j*cnt[x])%Mod)%Mod;
         }
     }
 }

 int main()
 {
     freopen("a.in","r",stdin);
     // freopen("me.out","w",stdout);
     while(scanf("%d",&n)!=EOF)
     {
         memset(son,,sizeof(son));
         memset(step,,sizeof(step));
         memset(pre,,sizeof(pre));
         memset(in,,sizeof(in));
         memset(cnt,,sizeof(cnt));
         memset(sum,,sizeof(sum));
         tot=;add_node();last=;
         for(int i=;i<=n;i++)
         {
             scanf("%s",s+);
             sl=strlen(s+);
             last=;
             for(int j=;j<=sl;j++)     extend(s[j]-'');
         }
         find_tp_1();//找拓扑序方法一 ： bfs=拓扑排序
         // find_tp_2();//方法二：直接对step[x]进行排序
         solve();
         ans=;
         for(int i=;i<=tot;i++) ans=(ans+sum[i])%Mod;
         printf("%d\n",ans);
         // cout<<"Time Used : "<<(double)clock()/CLOCKS_PER_SEC<<" s."<<endl;
     }
     return ;
 }