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--)
	{
		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));
		}
	}
}


©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页