HDU 1251 统计难题（字典树）

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1251

字典树就是这样一幅图

 

即每个结点有多个儿子结点。。。然后单词的前缀会在树中重合，

而结点的数据域随题目而变，例如这题，数据域prefixNum表示含以该结点结尾的前缀的单词的个数。

那么如果你想知道树中某单词是否存在，则数据域isWord == 1表示存在以该结点结尾的单词

 

字典树效率高在于他只需要遍历该单词

 

 

#include <iostream>
using namespace std;

struct Node
{
    struct Node *child[26];
    int perfixNum;//表示含以该结点结尾的前缀的单词的个数
};

Node *root;

void Init()//初始化
{
    root = new Node;
    for (int i = 0; i < 26; i++)
    {
        root->child[i] = NULL;
    }
}

//插入
void Insert(char word[])
{
    int len = strlen(word);
    Node *pNode = root;
    for (int i = 0; i < len; i++)
    {
        if (pNode->child[word[i] - 'a'] == NULL)
        {
            //不存在该结点，则创建一个
            Node *newNode = new Node;
            newNode->perfixNum = 1;
            for (int j = 0; j < 26; j++)
            {
                newNode->child[j] = NULL;
            }

            pNode->child[word[i] - 'a'] = newNode;
        }
        else
        {
            pNode->child[word[i] - 'a']->perfixNum++;
        }
        
        pNode = pNode->child[word[i] - 'a'];
    }
}

//查找，通过遍历该单词到该单词的结尾结点的prefixNum就是答案了
int Find(char word[])
{
    int len = strlen(word);
    Node *pNode = root;
    int i;
    for (i = 0; i < len; i++)
    {
        if (pNode->child[word[i] - 'a'] != NULL)
        {
            pNode = pNode->child[word[i] - 'a'];
        }
        else
        {
            break;
        }
    }

    if (i == len)
    {
        return pNode->perfixNum;
    }
    else
    {
        return 0;
    }
}

int main()
{

    char word[15], prefixWord[15];
    Init();
    while (gets(word), word[0] != 0)
    {
        Insert(word);
    }

    while (gets(prefixWord) != NULL)
    {
        printf("%d\n", Find(prefixWord));
    }
    return 0;
}