hdu1251（trie树）

Description

Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀). 

 

Input

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串. 

注意:本题只有一组测试数据,处理到文件结束. 

 

Output

对于每个提问,给出以该字符串为前缀的单词的数量. 

 

Sample Input

    
    

     
     banana
band
bee
absolute
acm

ba
b
band
abc 
    
    

 

Sample Output

    
    

     
     2
3
1
0 
    
    

这题的输入形式还挺有意思的，又get一招~

然后就是在这题的模板前提下，加了个num数组记录每个节点有多少单词经过以此判断前缀数量，所以输出这个num数组记录的数值便是前缀数量了~~

代码：

#include <iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
const int NODE = 1e6+10,CH = 26;

int ch[NODE][CH],sz,val[NODE],num[NODE];
char a1[100001][15],a2[15];
    /**
    ch[u][c]:   节点u指向的c儿子的边
    val[u]:     节点u的值
    sz:         trie树的size
    */
int idx(char c)
{return c-'a';}
int node()
{
        /**
        新建(初始化)节点
        **/
        memset(ch[sz],0,sizeof(ch[sz]));
        ///将所有儿子置为空
        val[sz]=0;
        return sz++;
}
void init()
{
        sz=0;
        ///trie树根节点为下标0的节点
        node();
}
void insert(char *s,int v)
{
        int u=0;
        num[u]++;
        for(;*s;s++)
        {

            int c=idx(*s);
            ///如果节点不存在 新建节点 并把值赋给当前节点的c儿子边
            if(!ch[u][c])
                ch[u][c]=node();
            ///继续移动

          //  cout<<u<<' '<<char(c+'a')<<' '<<ch[u][c]<<endl;
            u=ch[u][c];num[u]++;
        }
        ///在末尾节点记录信息
        val[u]=v;
}
int find(char *s)
{
    int u=0;
    for(;*s;s++)
    {
        int c=idx(*s);
       // cout<<u<<' '<<(char)(c+'a')<<' '<<num[u]<<endl;
        ///如果u节点没有c儿子  结束
            if(!ch[u][c])
            return 0;
            u=ch[u][c];
       // cout<<(char)*s<<endl;
    }
    return num[u];
}


int main()
{
    int o=0;
    memset(num,0,sizeof(num));
    init();
    while(gets(a1[o])&&a1[o][0])
    {
        insert(a1[o],o+1);
        o++;
    }
    while(gets(a2))
    {
        cout<<find(a2)<<endl;
    }


    return 0;
}