HDU 6138 Fleet of the Eternal Throne [AC自动机]

题意：给你n个串，m个询问，每次询问第x个串和第y个串，找到最长的满足，串z是x和y的公共子串并且是这n个串中某个串的前缀。

题解：AC自动机建树，标记每个点的长度，然后跑x和y的串，分别进行染色，找到相同染色点的地方，取ans的max。

AC代码：

#include<stdio.h>
#include<string.h>
#include<queue>
#include<string>
#define maxlen 100005
using namespace std;
struct Trie
{
int next[maxlen][30],fail[maxlen],end[maxlen],mark[maxlen],root,L;//next记录节点，在这里end指针代表以当前节点为字符串尾的字符串个数
int newnode()
{
for(int i=0;i<30;i++)
next[L][i]=-1;//节点连接的边初始化为-1
return L++;
}
void init()
{
L=0;
root=newnode();
}

void insert(char buf[])//trie树的建立
{
int l=strlen(buf);
int now=root;
for(int i=0;i<l;i++)
{
if(next[now][buf[i]-'a']==-1)next[now][buf[i]-'a']=newnode();
now=next[now][buf[i]-'a'];
end[now]=i+1;
}
}
void build()//建立ac自动机
{
queue<int>que;
for(int i=0;i<30;i++)
{
if(next[root][i]==-1)next[root][i]=root;
else //若有连边则将节点加入队列 ，并将fail指针指向root
{
fail[next[root][i]]=root;
que.push(next[root][i]);
}
}
while(!que.empty())
{
int now=que.front();
que.pop();
for(int i=0;i<30;i++)
{
if(next[now][i]==-1) //若无连边，则将该边指向当前节点fail指针指向的相应字符连接的节点
next[now][i]=next[fail[now]][i];
else //若有连边，则将儿子节点的fail指针指向当前节点fail指针指向相应字符接的节点
{
fail[next[now][i]]=next[fail[now]][i];
que.push(next[now][i]); //加入队列继续遍历
}
}
}
}
int query(char buf[],int flag)
{
int l=strlen(buf);
int now=root;
int ans=0;
for(int i=0;i<l;i++)
{
now=next[now][buf[i]-'a'];
int temp=now;
while(temp!=root)//根据题目要求改变形式
{
if(mark[temp])ans=max(ans,end[temp]);
if(flag==0)mark[temp]=1;
temp=fail[temp];
}
}
return ans; //在这里返回的是匹配到的模式串的数量
}
}ac;

string A[100005];
char buf[100005];
int main()
{
int T;
scanf("%d",&T);
while(T--
4000
)
{
ac.init();
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
A[i]="";
scanf("%s",buf);
int l=strlen(buf);
for(int j=0;j<l;j++)
A[i]+=buf[j];
ac.insert(buf);
}
ac.build();
int m;
scanf("%d",&m);
for(int i=0;i<m;i++)
{
memset(ac.mark,0,sizeof(ac.mark));
int x,y;
scanf("%d%d",&x,&y);
int l=A[x].size();
for(int j=0;j<l;j++)
buf[j]=A[x][j];
buf[l]='\0';
ac.query(buf,0);
l=A[y].size();
for(int j=0;j<l;j++)
buf[j]=A[y][j];
buf[l]='\0';
printf("%d\n",ac.query(buf,1));
}
}
}