hdu6208 The Dominator of Strings(AC自动机 模板题)

题目

t组样例，每次给你n个串，

问是否存在一个串，其它n-1个串都是它的子串

存在的话，输出这个串；否则输出No

思路来源

https://blog.csdn.net/qq_29480875/article/details/78013177

题解

mx[]存最长串，tmp[]存读入串

每次读入时进行比较，把短的插到trie树里去，

最后最长的mx串当模式串匹配，匹配到一个就将其清零，防止重复统计

如果能匹配到n-1个串，就有解；否则无解

注意用到这个节点的时候再清空的trie树写法，防TLE

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
//trie是节点 fail是指针 num是节点出现串的个数
//s是读入的要被插入trie树的串 t是要匹配的串
int t,n,cnt; 
int trie[maxn][26],fail[maxn],num[maxn];
char mx[maxn],tmp[maxn];//mx维护最长串 
int maxlen,nowlen;
void add(char s[],int len)
{
	int now=0;
	for(int j=0;j<len;++j)
	{
            int ch=s[j]-'a';
            if(!trie[now][ch])
	    {
		trie[now][ch]=++cnt;
		memset(trie[cnt],0,sizeof trie[cnt]);//要用了 现清 
	        num[cnt]=0;//要用了 现清 
	    }
            now=trie[now][ch];
        }
        num[now]++;
}
void build(int n)
{
	memset(trie[0],0,sizeof trie[0]);
	memset(fail,0,sizeof fail); 
        maxlen=nowlen=0;
	cnt=0;
        for(int i=1;i<=n;++i)
	{
            scanf("%s",tmp);
            nowlen=strlen(tmp);
		//超过最大的 就把非最大的插到trie树里 
            if(nowlen>=maxlen)
            {
        	if(maxlen>0)add(mx,maxlen);
        	maxlen=nowlen;
        	strcpy(mx,tmp);
            }
            else add(tmp,nowlen);
        }
}

void setFail()
{
    queue<int>q;
    for(int i=0;i<26;++i)
	if(trie[0][i])q.push(trie[0][i]);
	while(!q.empty())
	{
            int now=q.front();
	    q.pop();
            for(int i=0;i<26;++i)
            {
	        int &to=trie[now][i];
                if(to)
		{
			q.push(to);
			fail[to]=trie[fail[now]][i];
                }
		else trie[now][i]=trie[fail[now]][i];
            }
        }
}
bool solve(char s[],int len)
{
	int ans=0;
        for(int i=0,j=0;i<len;++i)
	{
            j=trie[j][s[i]-'a'];
            for(int temp=j;temp;temp=fail[temp])
            {
		ans+=num[temp];
		num[temp]=0;//出现过之后就清零 避免重复统计
    	    }
        }
        return ans==n-1; 
}
int main()
{
    	scanf("%d",&t);
    	while(t--)
        {
 		scanf("%d",&n);
                build(n);
	        setFail();
		if(!solve(mx,maxlen))puts("No");
		else printf("%s\n",mx);
   	}
 	    return 0;
}