hdu

最新推荐文章于 2019-01-02 14:22:55 发布

煎饼果子来一套

最新推荐文章于 2019-01-02 14:22:55 发布

阅读量491

点赞数

分类专栏： ACM 文章标签： tree insert null ini web

本文链接：https://blog.csdn.net/yobobobo/article/details/7894396

版权

ACM 专栏收录该内容

66 篇文章 0 订阅

订阅专栏

模版题。。。。注意只统计总类

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<vector>
#include<queue>
#define val 10005
using namespace std;

struct Dictree
{
	int cnt;
	int no;
	Dictree *next[100],*fail;
}*root;
vector<int> hack;
queue<Dictree *> q;
int n,m;
bool flag;
int mark[val];
char s[val],p[val];
void built_tree();
void set_fail();
void ac_auto(char *);
void empty_tree(Dictree *);
int main()
{
	int i,j,cnt;
	while(scanf("%d",&n)!=EOF)
	{
		getchar();
		built_tree();
		set_fail();
		cnt=0;
		scanf("%d",&m);
		getchar();
		for(i=1;i<=m;i++)
		{
			gets(s);
			hack.clear();
			flag=false;
			memset(mark,false,sizeof(mark));
			ac_auto(s);
			cnt+=flag;
			if(flag)
			{
				printf("web %d:",i);
				for(j=1;j<=n;j++)
					if(mark[j]) printf(" %d",j);
				putchar('\n');
			}
		}
		printf("total: %d\n",cnt);
		empty_tree(root);
	}
	return 0;
}
void built_tree()
{
	void insert(char *,int );
	int i;
	root=(Dictree*)malloc(sizeof(Dictree));
	root->no=0;
	root->fail=NULL;
	root->cnt=0;
	memset(root->next,NULL,sizeof(root->next));
	for(i=0;i<n;i++)
	{
		gets(s);
		insert(s,i+1);
	}
}
void insert(char *s,int key)
{
	int i;
	Dictree *t,*p=root;
	for(i=0;s[i];i++)
	{
		if(p->next[s[i]-32]==NULL)
		{
			t=(Dictree *)malloc(sizeof(Dictree));
			t->cnt=0;
			t->no=0;
			t->fail=NULL;
			memset(t->next,NULL,sizeof(t->next));
			p->next[s[i]-32]=t;
		}
		p=p->next[s[i]-32];
	}
	p->cnt++;
	p->no=key;
}
void set_fail()
{
	int i;
	Dictree *p,*temp;
	q.push(root);
	while(!q.empty())
	{
		p=q.front();
		q.pop();
		for(i=0;i<100;i++)
		{
			if(p->next[i])
			{
				if(p==root)   	p->next[i]->fail=root;
				else
				{
					temp=p->fail;
					while(temp)
					{
						if(temp->next[i])
						{
							p->next[i]->fail=temp->next[i];
							break;
						}
						temp=temp->fail;
					}
					if(temp==NULL) p->next[i]->fail=root;
				}
				q.push(p->next[i]);
			}
		}
	}
}
void ac_auto(char *s)
{
	int i,index;
	Dictree *p=root,*temp;
	for(i=0;s[i];i++)
	{
		index=s[i]-32;
		while(p!=root)
		{
			if(p->next[index]) break;
			p=p->fail;
		}
		temp=p=p->next[index];
		if(p==NULL) temp=p=root;
		while(temp)
		{
			if(temp->cnt)
			{
				flag=true;
				mark[temp->no]=true;
				hack.push_back(temp->no);
			}
			temp=temp->fail;
		}
	}
}
void empty_tree(Dictree *p)
{
	int i;
	for(i=0;i<26;i++)
		if(p->next[i])  empty_tree(p->next[i]);
	free(p);
}