Trie树(字典树)//洛谷P1481+洛谷P2580

第一行——咕咕咕。

第二行——慢慢学呗，还能离咋地。

 

Trie树：又称字典树，前缀树。是一种通过增加空间复杂度来减少时间复杂度的结构。其特点是根节点为空，其余节点存一个字符，节点与节点之间的路径是一个字符（一般是用于字符串）。

建树：令根节点为空，从该点出发，将新的字符串上的字符一个个加入，未出现过即直接相连，出现过找到最末尾的共享字符，连接即可。

性质：1.从根节点到某节点的路径即为一个字符串。2.每个节点的子节点所包含的字符串都不相同

其他：时间复杂度为（n+m）*l，n为字符串个数，m为查询个数，l为字符串长度

常用的操作有：插入，查询，删除。

先附题目链接：洛谷P1481 + 洛谷P2580

//P1481
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#define inf 0x3f3f3f3f
#define ll long long
#define maxn 2005
//字典树
using namespace std;

int trie[maxn][maxn];
int ans[maxn];//用于记录以i结尾的单词的个数
int tot = 0;
char c[maxn];//用于输入字符串

void init()
{
    for(int i = 0; i < maxn; ++i)
        for(int j = 0; j < maxn; ++j)
        {
            trie[i][j] = 0;
            ans[i] = 0;
        }
}//初始化

void insertS(char *c)
{
    int l = strlen(c);
    int u = 0;//代表根节点
    for(int i = 0; i < l; ++i)
    {
        int a = c[i] - 'a';//用int的形式储存
        if(trie[u][a] == 0)
            trie[u][a] = ++tot;//没有这个节点，新建
        u = trie[u][a];//跳到该节点
    }
    ans[u]++;//表示以u为结尾的单词的个数
}//建树

bool findS(int x)
{
    for(int i = 0; i < maxn; ++i)
    {
        if(trie[x][i])//从根节点开始，trie其实存的是编号
        {
            ans[trie[x][i]]+=ans[x];//更新ans为以trie[x][i]的单词数+以x为末尾
            //这样处理后我们只需找到叶子节点即可得出res
            //不过反正是求个max，遍历也可
            findS(trie[x][i]);
        }
    }
    return true;
}

int main()
{
    init();
    int n;
    scanf("%d",&n);
    while(n--)
    {
        scanf("%s",c);
        insertS(c);
    }
    findS(0);
    int res = 0;
    for(int i = 0; i < maxn; ++i)
        res = max(res,ans[i]);
    printf("%d\n",res);
    return 0;
}

//P2580
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#define inf 0x3f3f3f3f
#define ll long long
#define maxn 500005

using namespace std;

int trie[maxn][27];
int ans[maxn];
int tot = 0;
char c[maxn];

void insertS(char *c)
{
    int l = strlen(c);
    int u = 0;
    for(int i = 0; i < l; ++i)
    {
        int a = c[i] - 'a';
        if(trie[u][a] == 0)
            trie[u][a] = ++tot;
        u = trie[u][a];
    }
    ans[u]++;
}

int findS(char *c)
{
    int l = strlen(c);
    int u = 0;
    for(int i = 0; i < l; ++i)
    {
        int a = c[i] - 'a';
        if(trie[u][a] == 0) return 0;
        u = trie[u][a];
    }
    ans[u]++;
    return ans[u]-1;
}

int main()
{
    memset(trie,0,sizeof(trie));
    memset(ans,0,sizeof(ans));
    int n,m;
    scanf("%d",&n);
    while(n--)
    {
        scanf("%s",c);
        insertS(c);
    }
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s",c);
        int res = findS(c);
        if(res == 0) printf("WRONG\n");
        else if(res == 1) printf("OK\n");
        else printf("REPEAT\n");
    }
    return 0;
}

欢迎指出错误qwq