HDU 1251 统计难题 字典树-trie树

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

可以说是一道模板题。

代码如下：

#include <iostream>
#include <vector>
#include <List>
#include <deque>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <cmath>
using namespace std;

const int N=26;
typedef long long LL;

char xh[16];

struct Trie
{
    Trie *next[N];//next是表示每层有多少种类的数，如果只是小写字母，则26即可，若改为大小写字母，则是52，若再加上数字，则是62了，这里根据题意来确定。
    int v;//v可以表示一个字典树到此有多少相同前缀的数目，这里根据需要应当学会自由变化。
}*root;

void setroot()
{
    root=(Trie *)malloc(sizeof(Trie));
    for(int i=0;i<N;i++)
        root->next[i] = NULL;
}

void createTrie(char *str)
{
    int i,j,id;
    int len=strlen(str);
    Trie *p=root,*q;
    for(i=0;i<len;i++)
    {
        id=str[i]-'a';//这个地方依据题意来定
        if(p->next[id]==NULL)
        {
            q=(Trie *)malloc(sizeof(Trie));
            q->v=1;//如果没有的话，初始为1
            for(j=0;j<N;j++)
                q->next[j]=NULL;
            p->next[id]=q;
            p=p->next[id];
        }
        else
        {
            p->next[id]->v++;
            p=p->next[id];
        }
    }
}
int findTrie(char *str)
{
    int i,id;
    int len=strlen(str);
    Trie *p=root;
    for(i=0;i<len;i++)
    {
        id=str[i]-'a';
        p=p->next[id];
        if(p==NULL)     //若为空集，表示不存以此为前缀的串
            return 0;
    }
    return p->v;      //此串是字符集中某串的前缀
}

void dealTrie(Trie *T)
{//若果有多组数据-多棵树，那么要释放内纯
    int i;
    if(T==NULL)
        return ;
    for(i=0;i<N;i++)
        if(T->next[i]!=NULL)
            dealTrie(T->next[i]);
    free(T);//释放
}

int main()
{
    int i,j;
    setroot();
    while(gets(xh)&&xh[0]!='\0')
        createTrie(xh);
    memset(xh,0,sizeof(xh));
    while(gets(xh)!=NULL)
    {
        printf("%d\n",findTrie(xh));
    }
    return 0;
}