读书人

HDU-1251-统计难点

发布时间: 2012-10-27 10:42:26 作者: rapoo

HDU-1251-统计难题
trie树的最基本题型,题目是要统计在一个字典中,有多少匹配指定前缀的单词,因为trie树是一棵前缀树,所以对于trie树的每个节点在插入时只要经过这个节点便对这个计数变量加一,最后按照trie查询进行统计;返回统计量即可。


#include <iostream>#include <cstdio>#include <cstring>using namespace std;struct node{    int num;    bool f;    int next[26];    void init(){num=0;f=0;memset(next,-1,sizeof(next));};}trie[1000000];int p;char input[100];void root(){    trie[p=0].init();}void insert(char* a){    int index=0;    int len=strlen(a);    int cur;    for(int i=0;i<len;i++)    {        cur=a[i]-'a';        if(trie[index].next[cur]==-1)        {            trie[++p].init();            trie[index].next[cur]=p;        }        trie[index].num++;        index=trie[index].next[cur];    }    trie[index].f=1;    trie[index].num++;}int find(char* a){    int index=0;    int len=strlen(a);    int cur;    for(int i=0;i<len;i++)    {        cur=a[i]-'a';        if(trie[index].next[cur]==-1)return 0;        index=trie[index].next[cur];    }    return trie[index].num;}int main(){    char input[100];    int word;    int query;    root();    int i;    while(1)    {        cin.getline(input,20);        //scanf("%s",input);        if(strlen(input)==0)    break;        insert(input);    }    while(scanf("%s",input)!=EOF)    {        //scanf("%s",input);        cout<<find(input)<<endl;    }    return 0;}

读书人网 >编程

热点推荐