Trie树（字典树）自学、整理

之前就学过字典树，学的时候接触oi的时间还比较少，对字典树的理解还不是很透彻，那个时候还是别人手把手教我打出的字典树代码。然而，现在再看字典树，发现并不是那么难，反而觉得很简单了。

字典树其实是一种拿空间换时间的算法，对于做一些查找单词的题目来说还是很有用的。

下面以洛谷p2580为例子：

P2580 于是他错误的点名开始了

题目背景

XS中学化学竞赛组教练是一个酷爱炉石的人。

他会一边搓炉石一边点名以至于有一天他连续点到了某个同学两次，然后正好被路过的校长发现了然后就是一顿欧拉欧拉欧拉（详情请见已结束比赛CON900）。

题目描述

这之后校长任命你为特派探员，每天记录他的点名。校长会提供化学竞赛学生的人数和名单，而你需要告诉校长他有没有点错名。（为什么不直接不让他玩炉石。）

输入输出格式

输入格式：

第一行一个整数 n，表示班上人数。接下来 n 行，每行一个字符串表示其名字（互不相同，且只含小写字母，长度不超过 50）。第 n+2 行一个整数 m，表示教练报的名字。接下来 m 行，每行一个字符串表示教练报的名字（只含小写字母，且长度不超过 50）。

输出格式：

对于每个教练报的名字，输出一行。如果该名字正确且是第一次出现，输出“OK”，如果该名字错误，输出“WRONG”，如果该名字正确但不是第一次出现，输出“REPEAT”。（均不加引号）

输入输出样例

输入样例#1：  复制

5  
a
b
c
ad
acd
3
a
a
e

输出样例#1：  复制

OK
REPEAT
WRONG

说明

对于 40%的数据，n≤1000,m≤2000；

对于 70%的数据，n≤10000,m≤20000；

对于 100%的数据， n≤10000，m≤100000。

T1总是送分的。

其实这道题目我用map也过了，但是我还是写了一遍Trie

一、首先介绍一下Trie树的结构体。

struct node
{
    int sz;///表示以这个节点结束的单词个数，可以用来判断这里是否有单词或是记录这个单词出现的个数
    int isok;///判断这个单词有没有查找过，查找过就标成一，没有查过就标成零。
    node *son[26];///分支的26个字母，从a-z，不解释
    node()///清零（初始化）函数------其实用处不大。
    {
        for(int i=0; i<=25; i++)
        {
            son[i]=NULL;
        }
        sz=0;
        isok=0;
    }

};

二、插入单词的函数。

void my_insert()
{
    int n;///插入n个单词
    string ss;
    cin>>n;
    for(int ti=1;ti<=n;ti++)
    {
        cin>>ss;///插入的单词
        int l=ss.size();
        node *now=&start;///###重点###这里要注意
        for(int i=0; i<=l-1; i++)
        {
            if(now->son[ss[i]-'a']==NULL)///如果某个字母根本没出现过，把它补上，新建一个节点
            {
                now->son[ss[i]-'a']=new node;///ss[i]-'a'得到一个int，是指ss[i]对应的son
            }
            now=now->son[ss[i]-'a'];///向下一位推进
            if(i==l-1)///如果到了最后一位，把那一位的sz++，表示这里多了这样的一个单词
            {
                now->sz++;
            }
        }
    }

}

三、查找单词的函数。

void my_find()
{
    int m;

    cin>>m;///有m次询问

    for(int ti=1;ti<=m;ti++)
    {
        int flag;///记录最后判断的状态，flag等于-1表示不存在，等于0表示重复查找过，等于1表示第一次查找
        string ss;///需要判断的单词
        cin>>ss;
        int l=ss.size();
        node *now=&start;///依然是重点，容易出错。
        for(int i=0;i<=l-1;i++)
        {
            if(now->son[ss[i]-'a']==NULL)///如果某个字母根本没出现过，说明这个单词就不存在，把flag标成-1，然后退出循环
            {
                flag=-1;
                break;
            }
            now=now->son[ss[i]-'a'];//向下一位推进
            if(i==l-1)///到了最后一位
            {
                if(now->sz==0)///如果没有这样的单词，flag=-1，退出循环
                {
                    flag=-1;
                    break;
                }
                if(now->isok==1)///如果曾经查找过，flag=0，退出循环
                {
                    flag=0;
                    break;
                }
                else///其余情况，只有是没查找过且存在的，flag=1，把isok标成1，表示已经查找过，退出循环
                {
                    flag=1;
                    now->isok=1;///别忘了
                    break;
                }
            }
        }
        if(flag==-1)///没找到
            cout<<"WRONG"<<endl;
        if(flag==0)///重复
            cout<<"REPEAT"<<endl;
        if(flag==1)///第一次查找
            cout<<"OK"<<endl;
    }

}

这题就这两个函数就可以了，可以当做模板来学习（个人观点）

下面给出完整代码：

#include<iostream>
#include<cstdio>
using namespace std;

struct node
{
    int sz;
    int isok;
    node *son[26];
    node()
    {
        for(int i=0; i<=25; i++)
        {
            son[i]=NULL;
        }
        sz=0;
        isok=0;
    }
};
node start;
void my_insert()
{
    int n;
    string ss;
    cin>>n;
    for(int ti=1;ti<=n;ti++)
    {
        cin>>ss;
        int l=ss.size();
        node *now=&start;
        for(int i=0; i<=l-1; i++)
        {
            if(now->son[ss[i]-'a']==NULL)
            {
                now->son[ss[i]-'a']=new node;
            }
            now=now->son[ss[i]-'a'];
            if(i==l-1)
            {
                now->sz++;
            }
        }
    }
}
void my_find()
{
    int m;
    cin>>m;
    for(int ti=1;ti<=m;ti++)
    {
        int flag;
        string ss;
        cin>>ss;
        int l=ss.size();
        node *now=&start;
        for(int i=0;i<=l-1;i++)
        {
            if(now->son[ss[i]-'a']==NULL)
            {
                flag=-1;
                break;
            }


            now=now->son[ss[i]-'a'];
            if(i==l-1)
            {
                if(now->sz==0)
                {
                    flag=-1;
                    break;
                }
                if(now->isok==1)
                {
                    flag=0;
                    break;
                }
                else
                {
                    flag=1;
                    now->isok=1;
                    break;
                }
            }
        }
        if(flag==-1)
            cout<<"WRONG"<<endl;
        if(flag==0)
            cout<<"REPEAT"<<endl;
        if(flag==1)
            cout<<"OK"<<endl;
    }
}
int main()
{
    my_insert();
    my_find();
    return 0;
}