bzoj 5137: [Usaco2017 Dec]Standing Out from the Herd

题意：

给n个串，问每个串中有多少个子串只在这个串中出现，本质相同的串算一次。

题解：

显然可以用广义SAM做，一个串要贡献答案则要求它的right只在某个串中，所以对每个节点染色，最后统计一下就好了。
code：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
struct SAM{
    int par,max,a[26];
}sam[200010];int tot,root,ri[200010];
struct trnode{
    int a[26],c;
}tr[100010];int trtot=1;
struct node{
    int y,next;
}a[200010];int len=0,last[200010];
int n;LL ans[100010];
char s[100010];
void ins(int x,int y)
{
    a[++len].y=y;
    a[len].next=last[x];last[x]=len;
}
void update(int &c,int op)
{
    if(c==0) c=op;
    else c=(c==op)?c:-1;
}
int addsam(int p,int c,int op)
{
    int np=++tot;
    sam[np].max=sam[p].max+1;
    update(ri[np],op);
    for(;p&&!sam[p].a[c];p=sam[p].par) sam[p].a[c]=np;
    if(!p) sam[np].par=root;
    else
    {
        int q=sam[p].a[c];
        if(sam[q].max==sam[p].max+1) sam[np].par=q;
        else
        {
            int nq=++tot;
            sam[nq]=sam[q];sam[nq].max=sam[p].max+1;
            sam[q].par=sam[np].par=nq;
            for(;p&&sam[p].a[c]==q;p=sam[p].par) sam[p].a[c]=nq;
        }
    }
    return np;
}
void build(int op)
{
    int len=strlen(s+1),x=1;
    for(int i=1;i<=len;i++)
    {
        update(tr[x].c,op);
        int c=s[i]-'a';
        if(tr[x].a[c]==0) tr[x].a[c]=++trtot;
        x=tr[x].a[c];
    }
    update(tr[x].c,op);
}
void bt(int x,int c,int p)
{
    int np;
    if(x==1) np=1;
    else np=addsam(p,c,tr[x].c);
    for(int i=0;i<26;i++)
        if(tr[x].a[i]!=0) bt(tr[x].a[i],i,np);
}
void dfs(int x)
{
    for(int i=last[x];i;i=a[i].next)
    {
        int y=a[i].y;
        dfs(y);update(ri[x],ri[y]);
    }
    if(x!=1&&ri[x]!=-1)
        ans[ri[x]]+=(LL)sam[x].max-(LL)sam[sam[x].par].max;
}
int main()
{
    tot=root=1; 
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s+1);
        build(i);
    }
    bt(1,0,0);
    for(int i=2;i<=tot;i++) ins(sam[i].par,i);
    dfs(1);
    for(int i=1;i<=n;i++) printf("%lld\n",ans[i]);
}