杭电 HDU 2222 Keywords Search

/* THE PROGRAM IS MADE BY PYY */
/*----------------------------------------------------------------------------//
	Copyright (c) 2011 panyanyany All rights reserved.

	URL   : http://acm.hdu.edu.cn/showproblem.php?pid=2222
	Name  : 2222 Keywords Search

	Date  : Tuesday, 18 August, 2011
	Time Stage : Many hours .....

	Result:
4445107	2011-08-18 19:53:24 Accepted 2222 234MS	30152K 3051 B C++ pyy


	Test Data:


	Review:
第二次做了，感觉比第一次更容易理解了。不过还是出了很多错误
//----------------------------------------------------------------------------*/

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#define max(a, b) (((a) > (b)) ? (a) : (b))
#define min(a, b) (((a) < (b)) ? (a) : (b))

#define infinity    0x7f7f7f7f
#define minus_inf    0x80808080

#define MAXSIZE 1000009
#define LESSMAX	55

typedef struct tagNODE {
//	int fail ;
	int cnt ;
	struct tagNODE * child[26], * fail ;
} NODE ;

char model[MAXSIZE], pattn[LESSMAX] ;

int tcase, n ;
int stkPtr ;		// pointer for stack
int head, tial ;	// pointers for queue
int count ;			// match words count

NODE * tmp, * newNode, * childNode, * parntNode ;
NODE * stack[LESSMAX * 10000], * queue[LESSMAX * 10000] ;

#define root stack[0]

void makeTrie ()
{
	int i, j ;
	int len = strlen (pattn) ;

	tmp = stack[0] ;

	for (i = 0 ; i < len ; ++i)
	{
		j = pattn[i] - 'a' ;
		if (!tmp->child[j])
		{
			newNode			= (NODE *) calloc (1, sizeof (NODE)) ;
			stack[stkPtr++]	= newNode ;								// 注意1
			tmp->child[j]	= newNode ;
		}
		tmp = tmp->child[j] ;
//		printf ("%c: %d, cnt: %d\n", pattn[i], tmp, tmp->cnt) ;
	}
	++tmp->cnt ;
//	printf ("-------.%c, %d\n", pattn[i-1], tmp->cnt) ;
}

void makeFail ()
{
	int i, j ;

	head = tial = 0 ;

	for (i = 0 ; i < 26 ; ++i)
		if (root->child[i])
		{
			root->child[i]->fail = root ;
			queue[tial++] = root->child[i] ;
		}

	while (head < tial)
	{
		parntNode = queue[head++] ;
		for (i = 0 ; i < 26 ; ++i)
		{
			childNode = parntNode->child[i] ;
			if (childNode)
			{
				tmp = parntNode->fail ;
				while (tmp && !tmp->child[i])
					tmp = tmp->fail ;
				if (tmp)
					childNode->fail = tmp->child[i] ;
				else
					childNode->fail = root ;

				queue[tial++] = childNode ;
			}
		}
	}
}

void ACAutomation ()
{
	int i, j ;
	int len = strlen (model) ;
	NODE * tmpFail ;

	tmp = root ;

	for (i = 0 ; i < len ; ++i)
	{
		j = model[i] - 'a' ;
		while (!tmp->child[j] && tmp != root)
			tmp = tmp->fail ;
		tmp = (tmp->child[j]) ? tmp->child[j] : tmp ;

		// 在相同递减后缀并且未被匹配过的字符串间跳跃
		// 并且记录cnt 值，然后清空cnt 值，表示以此
		// 字符结尾的这个单词已经记录过了

		// 但要注意，用来跳跃的指针变量，由于要不断地
		// 改变，即通过fail 指针向根寻找，所以要另外
		// 使用一个临时变量，不能直接用tmp 否则在下
		// 一次的for循环就会没用了
		tmpFail = tmp ;	
		while (tmpFail->cnt)
		{
//			printf ("%c\n", model[i]) ;
			count += tmpFail->cnt ;
			tmpFail->cnt = 0 ;
			tmpFail = tmpFail->fail ;
		}
	}
}

void recycle ()
{
	while (stkPtr)
		free (stack[--stkPtr]) ;
}

int main ()
{
	int i, j ;
	while (scanf ("%d", &tcase) != EOF)
	{
		while (tcase--)
		{
			count		= 0 ;
			stkPtr		= 1 ;		// 如果用了0 ，在“注意1”
			// 里stack[0]的值会被覆盖掉，然后不断悲剧……
			stack[0]	= (NODE *) calloc (1, sizeof (NODE)) ;

			scanf ("%d", &n) ;
			for (i = 0 ; i < n ; ++i)
			{
				scanf ("%s", pattn) ;
				getchar () ;
				makeTrie () ;
			}
			scanf ("%s", model) ;		// 这一步经常忘
			makeFail () ;
			ACAutomation () ;
			printf ("%d\n", count) ;

			recycle () ;	// 释放内存
		}
	}
	return 0 ;
}

------------------------------------新旧世纪永恒的分割线---------------------------------------------------

纠结啊～郁闷啊～这题是所谓的 ac自动机 模版题 居然一直爆内存？！！试了我几天几十次啊！！纠结啊。悲剧啊。我对比了下人家的代码，感觉自己的数组明明开得比人家少啊！怎么还是爆了？一开始以为是用了 queue类 的问题，但后来改用静态数组，依旧爆！今天，没有办法了，全部改用 calloc 并且及时释放内存，不开结构体数组了，只开了两个结构体指针数组。好吧，辛辛苦苦，终于提交了……AC了，内牛满面啊～～～老天啊，您老人家终于肯看我一眼了么？

附上大牛的解题报告：

http://www.cppblog.com/mythit/archive/2009/07/30/80633.html
http://www.cnblogs.com/destinydesigner/archive/2009/10/15/1584191.html

http://www.dev-club.net/xiangxixinxi/22010072906013413/201102100439240.html

4075629	2011-06-18 14:40:59	Accepted	2222	203MS	30184K	2807 B	C++	pyy
4069611	2011-06-14 20:10:26	Memory Limit Exceeded	2222	62MS	55052K	2883 B	C++	pyy
4069609	2011-06-14 20:09:48	Memory Limit Exceeded	2222	31MS	33368K	2885 B	G++	pyy
4069607	2011-06-14 20:09:32	Memory Limit Exceeded	2222	62MS	55124K	2927 B	G++	pyy
4069605	2011-06-14 20:06:38	Memory Limit Exceeded	2222	62MS	55052K	2927 B	C++	pyy
4069604	2011-06-14 20:05:47	Compilation Error	2222	0MS	0K	2906 B	C++	pyy
4069602	2011-06-14 20:05:15	Memory Limit Exceeded	2222	62MS	55056K	2925 B	C++	pyy
4069451	2011-06-14 18:19:49	Runtime Error (ACCESS_VIOLATION)	2222	62MS	6728K	2944 B	C++	pyy
4069377	2011-06-14 17:27:08	Wrong Answer	2222	312MS	28464K	2642 B	C++	pyy
4069374	2011-06-14 17:25:54	Memory Limit Exceeded	2222	62MS	55064K	2640 B	C++	pyy
4069373	2011-06-14 17:25:42	Memory Limit Exceeded	2222	62MS	55064K	2640 B	C++	pyy
4069368	2011-06-14 17:23:55	Memory Limit Exceeded	2222	46MS	55056K	2576 B	G++	pyy
4069365	2011-06-14 17:21:47	Memory Limit Exceeded	2222	62MS	55056K	2589 B	G++	pyy
4069364	2011-06-14 17:21:36	Memory Limit Exceeded	2222	62MS	55068K	2589 B	C++	pyy
4069363	2011-06-14 17:20:12	Memory Limit Exceeded	2222	62MS	55068K	2589 B	C++	pyy

/* THE PROGRAM IS MADE BY PYY */ /*----------------------------------------------------------------------------// Copyright (c) 2011 panyanyany All rights reserved. URL : http://acm.hdu.edu.cn/showproblem.php?pid=2222 Name : 2222 Keywords Search Date : Saturday, 18 June, 2011 Time Stage : Many days ..... Result: Test Data: Review: //----------------------------------------------------------------------------*/ #include <stdio.h> #include <string.h> #include <stdlib.h> #define SIZE (50 * 10000 + 1) #define SIZE_OF_MODEL 1000010 typedef struct tagNODE { int cnt ; struct tagNODE *fail, *child[26] ; } NODE ; NODE *trie[SIZE], *root, *p, *queue[SIZE] ; char apattern[55], amodel[SIZE_OF_MODEL] ; int ikeyword, itcase, icount, ihead, itial, icursor ; void initial () { icursor = 0 ; icount = 0 ; root = (NODE*) calloc (1, sizeof (NODE)) ; } void recycle () { while (icursor) { free (trie[--icursor]) ; } free (root) ; } void makeTrie () { const int ciLen = strlen (apattern) ; int i, j, index ; p = root ; for (i = 0 ; i < ciLen ; ++i) { index = apattern[i] - 'a' ; if (! (p->child[index])) { trie[icursor] = (NODE *) calloc (1, sizeof (NODE)) ; p->child[index] = trie[icursor++] ; } p = p->child[index] ; } ++ p->cnt ; } void makeFail () { NODE *tmp, *tmpFail ; int i, j ; ihead = itial = 0 ; // root->fail = 0 ; for (i = 0 ; i < 26 ; ++i) { if (root->child[i]) { root->child[i]->fail = root ; queue[itial++] = root->child[i] ; } } while (ihead != itial) { tmp = queue[ihead++] ; // dequeue for (i = 0 ; i < 26 ; ++i) { if (tmp->child[i]) { queue[itial++] = tmp->child[i] ; // enqueue tmpFail = tmp->fail ; while (tmpFail) { if (tmpFail->child[i]) { tmp->child[i]->fail = tmpFail->child[i] ; break ; } tmpFail = tmpFail->fail ; } if (!tmpFail) { tmp->child[i]->fail = root ; } } } } } void acAutomation () { const int ciLen = strlen (amodel) ; int i, index ; NODE *tmp ; p = root ; i = 0 ; while (i < ciLen) { index = amodel[i] - 'a' ; while (p->child[index] == NULL && p != root) p = p->fail ; // printf ("/n1: %d/n", p->cnt) ; p = (p->child[index] == NULL) ? p : p->child[index] ; tmp = p ; while (tmp->cnt) { icount += tmp->cnt ; tmp->cnt = 0 ; tmp = tmp->fail ; // printf ("%d/t", icount) ; } ++i ; } } int main () { int i, j ; while (scanf ("%d", &itcase) != EOF) { while (itcase--) { initial () ; scanf ("%d", &ikeyword) ; while (ikeyword--) { scanf ("%s", apattern) ; makeTrie () ; } makeFail () ; scanf ("%s", amodel) ; acAutomation () ; printf ("%d/n", icount) ; recycle () ; } } return 0 ; }