UVA 1449 Dominating Patterns(AC自动机)

UVA 1449 Dominating Patterns(AC自动机)

http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=36265

题意：

        给你n个单词，还有一个长文本s，现在要你输出在文本s中出现次数最多的单词。

分析：

        直接建立AC自动机，并且val值是当前插入单词的编号。建立AC自动机后，用s文本串去匹配，每当走过val非0的节点后都使得对应单词的cnt值+1.最后一遍扫描cnt数组，然后求出max值。然后在扫描一遍输入单词，只要当前单词的cnt值==max就输出当前单词。

        注意：这里有个问题。如果输入单词有重复的情况如：

       单词aaa输入了两遍的话，那么在建立Trie的时候后来的aaa覆盖了前一个aaa，最终算得cnt[i]=0，cnt[i+1]=5。结果我们就只输出了一个aaa。其实我们应该输出两次aaa（在最终结果里）。这里我们应该做个map映射，将字符串=="aaa"的所有单词始终映射到序号x上，最终cnt[x]的值=5=max，最后我们扫描一遍单词aaa，用过map[aaa]来获得aaa的序号为x,这样就可以输出2遍aaa了。

AC代码：

#include<cstdio>
#include<queue>
#include<cstring>
#include<map>
#include<string>
#include<iostream>

using namespace std;
const int maxnode=11000;
const int sigma_size=26;
map<string,int >ms;

struct AC_Automata
{
	int ch[maxnode][sigma_size];
	int val[maxnode];
	int f[maxnode];
	int last[maxnode];
	int sz;
	int cnt[200];//计算字符串出现的次数
	void init()
	{
		sz=1;
		memset(ch[0],0,sizeof ch[0]);
		memset(cnt,0,sizeof cnt);
		ms.clear();
	}
	void insert(char *s,int v)//v是单词s输入的序号
	{
		int n=strlen(s),u=0;
		for(int i=0;i<n;i++)
		{
			int id=s[i]-'a';
			if(ch[u][id]==0)
			{
				ch[u][id]=sz;
				memset(ch[sz],0,sizeof ch[sz]);
				val[sz++]=0;
			}
			u=ch[u][id];
		}
		val[u]=v;
		ms[string(s)]=v;
	}
	void print(int i)
	{
		if(val[i])
		{
			cnt[val[i]]++;
			print(last[i]);
		}
	}
	void find(char *s)
	{
		int n=strlen(s);
		int j=0;

		for(int i=0;i<n;i++)
		{
			int id=s[i]-'a';
			while(j&&ch[j][id]==0)j=f[j];
			j=ch[j][id];
			if(val[j])print(j);
			else if(val[last[j]])
				print(last[j]);
		}
	}
	void getfail()
	{
		queue<int >q;
		last[0]=f[0]=0;
		for(int i=0;i<sigma_size;i++)
		{
			int u=ch[0][i];
			if(u)
			{
				f[u]=last[u]=0;
				q.push(u);
			}
		}
		while(!q.empty())
		{
			int r=q.front();
			q.pop();
			for(int i=0;i<sigma_size;i++)
			{
				int u=ch[r][i];
				if(!u)continue;
				q.push(u);
				int v=f[r];
				while(v&&ch[v][i]==0)v=f[v];
				f[u]=ch[v][i];
				last[u]=val[f[u]]?f[u]:last[f[u]];
			}
		}
	}
}ac;
char word[160][80];
char text[1000000+1000];
int main()
{
	int n;
	while(~scanf("%d",&n)&&n)
	{
		ac.init();
		for(int i=1;i<=n;i++)
		{
			scanf("%s",word[i]);
			ac.insert(word[i],i);
		}
		//cout<<"+++"<<endl;
		ac.getfail();
		scanf("%s",text);
		ac.find(text);
		int maxe=-1;
		for(int i=1;i<=n;i++)
		{
			if(ac.cnt[i]>maxe)
				maxe=ac.cnt[i];
		}

		printf("%d\n",maxe);
		for(int i=1;i<=n;i++)
		{
			if(ac.cnt[ms[string(word[i])]]==maxe)
				printf("%s\n",word[i]);
		}
	}
}