字典树03

    在经历了01和02后，终于可以独立写出字典树了。

hdoj: http://acm.hdu.edu.cn/showproblem.php?pid=1671

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
struct Tree
{
	int flag;
	Tree *next[10];
} *root;
void ptcre(char str[])
{
	Tree *s = root, *t;
	for(int i = 0; i < strlen(str); i++)
	{
		if(!s -> next[str[i] - '0'])
		{
			t = (Tree*) malloc(sizeof(Tree));
			memset(t, 0, sizeof(Tree));
			t -> flag = 0;
			s -> next[str[i] - '0'] = t;
			s = s -> next[str[i] - '0'];
		}
		else
			s = s -> next[str[i] - '0'];
	}
	s -> flag = 1;
}
int findans(char str[])
{
	Tree *s = root;
	for(int i = 0; i < strlen(str) - 1; i++)//注意遍历时不经过字符串最后一个节点，因为最后一个节点的flag = 1，如果经历会判断出错
	{
		s = s -> next[str[i] - '0'];
		if(s -> flag)
			return 0;
	}
	return 1;
}
void freedom(Tree *p)//注意动态分配内存，删除字典树就是这么简单
{
	int i;
	for(i = 0; i < 10; i++)
		if(p -> next[i])
			freedom(p -> next[i]);
	free(p);
}
main()
{
	int n, m;
	char nums[50000][11];
	scanf("%d", &n);
	while(n--)
	{
		root = (Tree*) malloc(sizeof(Tree));
		int k = 1;
		memset(root, 0, sizeof(Tree));
		scanf("%d", &m);
		for(int i = 0; i < m; i++)
		{
			scanf("%s", nums[i]);
			ptcre(nums[i]);
		}
		for(int i = 0; i < m; i++)
			if(!findans(nums[i]))
			{
				k = 0;
				puts("NO");
				break;
			}
		if(k)
			puts("YES");
		freedom(root);
	}
}