HDU 1251 统计难题

转载请注明出处：忆梦http://blog.csdn.net/yimeng2013/article/details/9698543

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1251

题目大意：找出此前缀的单词有多少个

题解：标准的字典树（Trie树），直接套用模板就可以AC了

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define MAX 26
typedef struct Trie
{
	Trie *next[MAX];
	int v;
};
Trie root;
void createTrie(char *str)  
{  
    int len = strlen(str);  
    Trie *p = &root, *q;  
    for(int i=0; i<len; ++i)  
    {  
        int id = str[i]-'a';  
        if(p->next[id] == NULL)  
        {  
            q = (Trie *)malloc(sizeof(Trie));  
            q->v = 1;    //初始v==1  
            for(int j=0; j<MAX; ++j)  
                q->next[j] = NULL;  
            p->next[id] = q;  
            p = p->next[id];  
        }  
        else  
        {  
            p->next[id]->v++;  
            p = p->next[id];  
        }  
    }  
    // p->v = -1;   //若为结尾，则将v改成-1表示(视情况而定)  
}  

int findTrie(char *str)  
{  
    int len = strlen(str);  
    Trie *p = &root;  
    for(int i=0; i<len; ++i)  
    {  
        int id = str[i]-'a';  //根据需要选择是减去'0'还是'a'，或者是'A'  
        p = p->next[id];  
        if(p == NULL)   //若为空集，表示不存以此为前缀的串  
            return 0;  
    }  
	return p->v;
 
}  
int main ()
{
	char str[12];
	int i;
	for(i = 0; i < MAX; i++)
		root.next[i] = NULL;

	while(gets(str) && str[0]!='\0')
		createTrie(str);

	memset(str, 0, sizeof(str));
	while(scanf("%s", str) != EOF)
	{
		int ans = findTrie(str);
		printf("%d\n", ans);
	}
	return 0;
}