bzoj3926 广义后缀自动机

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=3926

大意：给一棵树，每个节点代表一个字符，任意2个节点之间可以表示为一个字符串，问不同字符串的个数。（数据保证叶子节点<=20）

思路同bzoj2780

先吐槽一下，这3题建边简直要炸了，幸好学到了种新姿势，在结构体里打内置函数（用到外面的变量名 ::变量名），这种题错了完全不想看，幸好没出什么傻逼问题。

回归正题

从叶子节点开始遍历，答案就是遍历出的很多字符串的子串的不同个数（好绕口。。。）。

加到广义后缀自动机中，相同父亲的节点加入时的last也是相同的。（我一开始竟然神经病的将它们加入到trie中，再加到广义后缀自动机中。。。。。。）

统计答案的时候，第一次我是按照以前做的题的思路，先更新len大的，统计它可以走到哪几个不同串。答案就是根节点的值-1；后来%了下神犇的代码。发现只要统计len[x]-len[fa[x]]的值就好了（蒟蒻终究还是太弱了。。。）

%完神犇后的代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,Co,Col[100040],Ru[100050];

struct SAM {
int son[17],len,fa;
}C[3000050];
int tot,last,V[3000040],q[3000040];
long long sum[3000040],ans;

inline void Add(int x,int &l)
{
int p=l;
if (C[p].son[x]) {
int q=C[p].son[x];
if (C[p].len+1==C[q].len) l=q;
else {
int nq=++tot;   C[nq]=C[q];
C[nq].len=C[p].len+1;
C[q].fa=nq;
for (;p&&C[p].son[x]==q;p=C[p].fa) C[p].son[x]=nq;
l=nq;
}
} else {
int np=++tot;   l=tot;
C[np].len=C[p].len+1;
for (;p&&!C[p].son[x];p=C[p].fa) C[p].son[x]=np;
if (!p) C[np].fa=1;
else {
int q=C[p].son[x];
if (C[p].len+1==C[q].len) C[np].fa=q;
else {
int nq=++tot; C[nq]=C[q];
C[nq].len=C[p].len+1;
C[q].fa=C[np].fa=nq;
for (;p&&C[p].son[x]==q;p=C[p].fa) C[p].son[x]=nq;
}
}
}
}
inline void Pre()
{
for (int i=1;i<=tot;++i) ++V[C[i].len];
for (int i=1;i<=tot;++i) V[i]+=V[i-1];
for (int i=tot;i;--i) q[V[C[i].len]--]=i;
int x,y;
for (int i=tot;i;--i) {
x=q[i]; sum[x]=1;
for (int j=0;j<Co;++j)
if (y=C[x].son[j])
sum[x]+=sum[y];
}
//  for (int i=2;i<=tot;++i) ans+=C[i].len-C[C[i].fa].len;
}
struct Trie {
int son[14];
}T[3000040];
int Tsum;
void Dfs(int x,int l)
{
int y,r;
for (int i=0;i<Co;++i)
if (y=T[x].son[i]) {
r=l;
Add(i,r);

4000
Dfs(y,r);
}
}
struct sef { int ne,en; };
struct Graph {
sef h[200050];
int tot,first[100050];
inline void setl(int x,int y) {
h[++tot].ne=first[x]; h[tot].en=y;
first[x]=tot;
}
void Dfs(int x,int fa,int rt) {
int y;
if (!T[rt].son[Col[x]]) T[rt].son[Col[x]]=++Tsum;
int ret=T[rt].son[Col[x]];
for (int i=first[x];y=h[i].en,i;i=h[i].ne)
if (y!=fa)  Dfs(y,x,ret);
}
}G;
int main()
{
scanf("%d%d",&n,&Co);
for (int i=1;i<=n;++i) scanf("%d",Col+i);
int x,y;
for (int i=1;i<n;++i) {
scanf("%d%d",&x,&y);
++Ru[x]; ++Ru[y];
G.setl(x,y);
G.setl(y,x);
}
for (int i=1;i<=n;++i)
if (Ru[i]==1) G.Dfs(i,0,0);
tot=last=1;
Dfs(0,1);
Pre();
printf("%lld\n",sum[1]-1);
}

加到trie中的傻逼代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,Co,Col[100040],Ru[100050];

struct SAM {
int son[17],len,fa;
}C[3000050];
int tot,last,V[3000040],q[3000040];
long long sum[3000040],ans;

inline void Add(int x,int &l)
{
int p=l;
if (C[p].son[x]) {
int q=C[p].son[x];
if (C[p].len+1==C[q].len) l=q;
else {
int nq=++tot;   C[nq]=C[q];
C[nq].len=C[p].len+1;
C[q].fa=nq;
for (;p&&C[p].son[x]==q;p=C[p].fa) C[p].son[x]=nq;
l=nq;
}
} else {
int np=++tot;   l=tot;
C[np].len=C[p].len+1;
for (;p&&!C[p].son[x];p=C[p].fa) C[p].son[x]=np;
if (!p) C[np].fa=1;
else {
int q=C[p].son[x];
if (C[p].len+1==C[q].len) C[np].fa=q;
else {
int nq=++tot; C[nq]=C[q];
C[nq].len=C[p].len+1;
C[q].fa=C[np].fa=nq;
for (;p&&C[p].son[x]==q;p=C[p].fa) C[p].son[x]=nq;
}
}
}
}
inline void Pre()
{
for (int i=1;i<=tot;++i) ++V[C[i].len];
for (int i=1;i<=tot;++i) V[i]+=V[i-1];
for (int i=tot;i;--i) q[V[C[i].len]--]=i;
int x,y;
for (int i=tot;i;--i) {
x=q[i]; sum[x]=1;
for (int j=0;j<Co;++j)
if (y=C[x].son[j])
sum[x]+=sum[y];
}
//  for (int i=2;i<=tot;++i) ans+=C[i].len-C[C[i].fa].len;
}
struct Trie {
int son[14];
}T[3000040];
int Tsum;
void Dfs(int x,int l)
{
int y,r;
for (int i=0;i<Co;++i)
if (y=T[x].son[i]) {
r=l;
Add(i,r);
Dfs(y,r);
}
}
struct sef { int ne,en; };
struct Graph {
sef h[200050];
int tot,first[100050];
inline void setl(int x,int y) {
h[++tot].ne=first[x]; h[tot].en=y;
first[x]=tot;
}
void Dfs(int x,int fa,int rt) {
int y;
if (!T[rt].son[Col[x]]) T[rt].son[Col[x]]=++Tsum;
int ret=T[rt].son[Col[x]];
for (int i=first[x];y=h[i].en,i;i=h[i].ne)
if (y!=fa)  Dfs(y,x,ret);
}
}G;
int main()
{
scanf("%d%d",&n,&Co);
for (int i=1;i<=n;++i) scanf("%d",Col+i);
int x,y;
for (int i=1;i<n;++i) {
scanf("%d%d",&x,&y);
++Ru[x]; ++Ru[y];
G.setl(x,y);
G.setl(y,x);
}
for (int i=1;i<=n;++i)
if (Ru[i]==1) G.Dfs(i,0,0);
tot=last=1;
Dfs(0,1);
Pre();
printf("%lld\n",sum[1]-1);
}