统计难题

hdoj1251

题目大意： 求出以某字符串为前缀的单词的数量

解决：trie树 

#include <iostream>

#include <cstring>

using namespace std;

struct node

{

    int cnt;

    int next[26];

};

node trie[3000000];

int top=0;

void init()

{//初始化字典树，0号为根节点，只需将next值置为0就行了

   memset(trie[0].next,0,sizeof(trie[0].next));

    top=1;

}



void insert(char *str)

{

    int i=0,id;//i为根节点

    while(*str)

    {

        id=*str-'a';

        if(trie[i].next[id]==0)

        {//若为空，开辟新空间，并初始化新空间

            trie[i].next[id]=top;

            trie[top].cnt=1;

            memset(trie[top].next,0,sizeof(trie[top].next));

            top++;

            i=trie[i].next[id];

        }

        else 

        {//若不空，指向下一个地址，并对下一个地址的元素操作

            i=trie[i].next[id];

            trie[i].cnt++;

        }

        str++;

    }

}



int search(char *str)

{

    int i=0,id;

    while(*str)

    {

        id=*str-'a';

        if(trie[i].next[id]==0)return 0;

        i=trie[i].next[id];

        str++;

    }

    return trie[i].cnt;

}



int main()

{

    char str[15];

    init();

    while(gets(str),strcmp(str,"")!=0)insert(str);

    while(~scanf("%s",str))printf("%d\n", search(str));

 //   system("pause");

    return 0;

}