HDU 1251 统计难题（字典树）

 

题目大意

 

给了很多单词，现在要统计以某个字符串为前缀的单词的数量

 

做法分析

 

直接建字典树，初始化出每个节点被覆盖的次数，然后对于每一个字符串，直接统计即可

 

参考代码

 

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

struct Tire
{
    struct Node
    {
        Node *next[26];
        int cnt;
        Node()
        {
            for(int i=0; i<26; i++) next[i]=NULL;
            cnt=0;
        }
    };
    Node *root;
    Tire()
    {
        root=new Node();
    }

    void Insert(char buff[])
    {
        Node *u=root;
        for(int i=0; buff[i]; i++)
        {
            int v=buff[i]-'a';
            if(u->next[v]==NULL) u->next[v]=new Node();
            u=u->next[v];
            u->cnt++;
        }
    }

    int Find(char buff[])
    {
        Node *u=root;
        for(int i=0; buff[i]; i++)
        {
            int v=buff[i]-'a';
            if(u->next[v]==NULL) return 0;
            u=u->next[v];
        }
        return u->cnt;
    }

    void Free(Node *u)
    {
        for(int i=0; i<26; i++)
            if(u->next[i]!=NULL) Free(u->next[i]);
        delete u;
    }
} tree;

char str[15];

int main()
{
    while(gets(str), strcmp(str, "")!=0) tree.Insert(str);
    while(gets(str)) printf("%d\n", tree.Find(str));
    tree.Free(tree.root);
    return 0;
}

AC通道

HDU 1251 统计难题