`

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;
}
0
2
分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics