标签:bzoj3926 zjoi2015 诸神眷顾的幻想乡 广义后缀自动机 后缀自动机
#include <stdio.h>
int main()
{
puts("转载请注明出处[vmurder]谢谢");
puts("网址:blog.csdn.net/vmurder/article/details/44891009");
}
http://wjmzbmr.com/archives/zjoi-2015-day-1%e9%a2%98%e8%a7%a3/26
首先我们发现叶子节点很少,所以可以从每个叶子节点开始扫,然后就建立起了一棵大大的Trie树!然后统计有多少子串,就需要把Trie建立成广义(多串)后缀自动机。
然后怎么统计呢?
为什么呢?
考虑abc建立起来的串,答案是6……嗯什么也发现不了。
然后abcb,我们发现第一个b字符和第二个b字符,它们的pa都是那个辅助节点。
发现是每个节点,从1开始到它的dep,都能构成长短不一的子串,但是其中或许会有重复,而对于这两个’b’,重复的部分都是pa的1~dep一部分,然后都减去了。
最终得到正确答案9。
#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 101000
#define T 10
using namespace std;
struct Eli
{
int v,next;
}e[N<<1];
int head[N],cnt,d[N];
inline void add(int u,int v)
{
d[v]++;
e[++cnt].v=v;
e[cnt].next=head[u];
head[u]=cnt;
}
int n,m,src[N];
struct SAM
{
int son[N*20][T],pa[N*20],dep[N*20]; // 那个吧。这里开少了一倍TAT,竟然AC了
int root,cnt,crs[N*20]; // 其实也不算奇怪,均摊下来差不多啦~(不是太好构造~)
void begin(){crs[0]=root=cnt=1;}
int newnode(int last)
{
dep[++cnt]=dep[last]+1;
return cnt;
}
int add(int p,int x)
{
int np=son[p][x];
if(np&&dep[np]==dep[p]+1)return np;
else np=newnode(p); /* 这是与单串后缀自动机唯一有区别的部分*/
while(p&&!son[p][x])son[p][x]=np,p=pa[p];
if(!p)pa[np]=root;
else {
int q=son[p][x];
if(dep[q]==dep[p]+1)pa[np]=q;
else {
int nq=newnode(p);
pa[nq]=pa[q],pa[q]=pa[np]=nq;
memcpy(son[nq],son[q],sizeof son[q]);
while(p&&son[p][x]==q)son[p][x]=nq,p=pa[p];
}
}
return np;
}
void dfs(int x,int p)
{
crs[x]=add(crs[p],src[x]);
int i,v;
for(i=head[x];i;i=e[i].next)
{
v=e[i].v;
if(v==p)continue;
dfs(v,x);
}
}
bool vis[N*20];
queue<int>q;
long long bfs()
{
long long ret=0;
q.push(root);
int i,u,v;
while(!q.empty())
{
u=q.front(),q.pop();
if(u!=root)ret+=dep[u]-dep[pa[u]];
for(i=0;i<m;i++)
{
v=son[u][i];
if(vis[v])continue;
vis[v]=true,q.push(v);
}
}
return ret;
}
}sam;
int main()
{
freopen("test.in","r",stdin);
int i,a,b;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)scanf("%d",&src[i]);
for(i=1;i<n;i++)
{
scanf("%d%d",&a,&b);
add(a,b),add(b,a);
}
sam.begin();
for(i=1;i<=n;i++)if(d[i]==1)
{sam.dfs(i,0);break;}
cout<<sam.bfs()<<endl;
return 0;
}
【BZOJ3926】【Zjoi2015】诸神眷顾的幻想乡 广义后缀自动机
标签:bzoj3926 zjoi2015 诸神眷顾的幻想乡 广义后缀自动机 后缀自动机
原文地址:http://blog.csdn.net/vmurder/article/details/44891021