在经历了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);
}
}