HUSTOJ 1609 AntiVirus(AC自动机)

题目链接：http://acm.hust.edu.cn/problem.php?id=1609

题意：给出n个短串和一个长串，求长串中包含了多少个短串？

思路：首先将长串变短，因为每个字母连续出现的次数最多是n个串中该字母最长的次数。然后将n个串建立自动机。

const int MAX=1000005;

int C;

char s[MAX],s1[MAX*150],str[2005][200];







struct Node

{

	int next[26],pre,leaf,flag;



	void init()

	{

		memset(next,-1,sizeof(next));

		pre=-1;

		leaf=-1;

		flag=0;

	}

};







Node a[600005];

int e,root,n;

int Q;



void insert(char str[],int id)

{

	int p=root;

	for(int i=0;str[i];i++)

	{

		if(a[p].next[str[i]-'a']==-1) a[p].next[str[i]-'a']=++e;

		p=a[p].next[str[i]-'a'];

	}

	a[p].leaf=id;

}





void build_DFA()

{

	int i,p,q,k;

	queue<int> Q;

	Q.push(1);

	while(!Q.empty())

	{

		p=Q.front();

		Q.pop();

		for(i=0;i<26;i++) if(a[p].next[i]!=-1)

		{

			q=a[p].next[i];

			k=a[p].pre;

			while(k!=-1)

			{

				if(a[k].next[i]!=-1)

				{

					a[q].pre=a[k].next[i];

					break;

				}

				else k=a[k].pre;

			}

			Q.push(q);

		}

	}

}



int ans[2005];



void find(char str[])

{

	int p=root,i,pre,k;

	for(i=0;str[i];i++)

	{

		k=str[i]-'a';

		while(p>=1)

		{

			if(a[p].next[k]!=-1)

			{

				p=a[p].next[k];

				pre=p;

				while(pre>=1&&a[pre].flag==0)

				{

					a[pre].flag=1;

					if(a[pre].leaf!=-1) ans[a[pre].leaf]=1;

					pre=a[pre].pre;

				}

				break;

			}

			else p=a[p].pre;

		}

		if(p<1) p=1;

	}

}



void init()

{

	int i;

	for(i=0;i<=600000;i++) a[i].init();

	for(i=0;i<26;i++) a[0].next[i]=1;

	a[1].pre=0;

}



void deal()

{

    int L=0,i,j,k,t,len,p[26]={0};

    for(i=1;i<=n;i++)

    {

        len=strlen(str[i]);

        for(j=0;j<len;j++)

        {

            k=1;

            for(t=j+1;t<len&&str[i][t]==str[i][j];t++)

            {

                k++;

            }

            p[str[i][j]-'a']=max(p[str[i][j]-'a'],k);

            j=t-1;

        }

    }

    len=strlen(s);

    for(i=0;i<len;i++)

    {

        if(s[i]>='a'&&s[i]<='z') s1[L++]=s[i];

        else

        {

            k=0;

            for(j=i+1;s[j]>='0'&&s[j]<='9';j++)

            {

                k=k*10+s[j]-'0';

            }

            i=j+1;

            for(j=1;j<=p[s[i-1]-'a']+1&&j<=k;j++) s1[L++]=s[i-1];

        }

    }

    s1[L]=0;

}



int main()

{

    for(scanf("%d",&C);C--;)

    {

        scanf("%d",&n);

        int i;

        init();

        root=e=1;

        for(i=1;i<=n;i++) scanf("%s",str[i]),insert(str[i],i);

        scanf("%s",s);

        deal();

        build_DFA();

        memset(ans,0,sizeof(ans));

        find(s1);

        int A=0;

        for(i=1;i<=n;i++) if(ans[i]) A++;

        if(A) printf("Yes");

        else printf("No");

        printf(" %d\n",A);

    }

    return 0;

}