字典树——统计难题 ( HDU 1251 )

1. 字典树：

   • 定义：
     字典树（Trie树）是一种树形数据结构，其作用是存储多个字符串，并可以自动按照字典序排好。该算法的时间复杂度为O(n)，空间复杂度为O(nk)，其中n为所有字符串的和，k为所有可能出现的字符的个数。

   • 实现方法：
     树上的每个节点代表了一个字符串，每个节点最多有k个孩子，第k个孩子代表的字符串为原结点的字符串后接上第k个字符构成。

   • Node结点代码：

struct node
{
    int num;
    node *son[26];
    node()
    {
        num=0;
        int i;
        for(i=0;i<26;i++)
            son[i]=NULL;
    }
};

• 插入字符串代码：

void insert(char *str)
{
    int i,m;
    cur=root;
    for(i=0;i<strlen(str);i++)
    {
        m=str[i]-'a';
        if(cur->son[m]!=NULL)
        {
            cur=cur->son[m];
            ++(cur->num);
        }
        else
        {
            NewNode=new node;
            ++(NewNode->num);
            cur->son[m]=NewNode;
            cur=NewMode;
        }
    }
}

• 查询操作代码：

int search(char *str)
{
    int i,m;
    cur=root;
    for(i=0;i<strlen(str);i++)
    {
        m=str[i]-'a';
        if(cur->son[m]==NULL)
            return 0;
        cur=cur->son[m];
    }
    return cur->num;
}

• 初始化元素：

node * root = new node;
node * cur  = new node;
node * NewNode = new node;

1. 练习：

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

   • 分析&题解：
     非常裸的字典树，外加一个查找计算的函数即可。

   • AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
struct node
{
    int count;
    node *childs[26];
    node()
    {
        count=0;
        int i;
        for(i=0;i<26;i++)
        childs[i]=NULL;
    }
};
node *root=new node;
node *current,*newnode;
void insert(char *str)
{
    int i,m;
    current=root;
    for(i=0;i<strlen(str);i++)
    {
        m=str[i]-'a';
        if(current->childs[m]!=NULL)
        {
            current=current->childs[m];
            ++(current->count);
        }
        else
        {
            newnode=new node;
            ++(newnode->count);
            current->childs[m]=newnode;
            current=newnode;
        }
    }
}
int search(char *str)
{
    int i,m;
    current=root;
    for(i=0;i<strlen(str);i++)
    {
        m=str[i]-'a';
        if(current->childs[m]==NULL)
        return 0;
        current=current->childs[m];
    }
    return current->count;
}
int main()
{
    char str[20];
    while(gets(str),strcmp(str,""))
    insert(str);
    while(gets(str)!=NULL)
    printf("%d\n",search(str));
    return 0;
}