字符串--P8306 【模板】字典树

一.什么是字典树

Trie 树，即字典树，是一种树形结构。典型应用是用于统计和排序大量的字符串前缀来减少查询时间，最大限度地减少无谓的字符串比较。

Trie 树的核心思想是空间换时间。利用字符串的公共前缀来降低查询时间的开销以达到提高效率的目的。

二.字典树的性质

1.根节点不包含字符，除根节点外每一个节点都只包含一个字符。

2.从根节点到某一节点，路径上经过的字符连接起来，为该节点对应的字符串。

3.每个节点的所有子节点包含的字符都不相同。

下图就是一个字典树：

---

---

---

来看看具体题面吧 

 

#include <bits/stdc++.h>
using namespace std;

const int N = 3000005;

int son[N][70], cnt[N], idx;
// 0号点既是根节点，又是空节点
// son[][]存储树中每个节点的子节点序号
// cnt[]存储以每个节点结尾的单词数量

int translate(char x)   //映射字符
{
    int temp;
    if(x >= '0' && x <= '9') temp = x - '0';  // 数字就映射成本身
    if(x >= 'A' && x <= 'Z') temp = x - 'A' + 10;  //大写字母映射成 10 到  35
    if(x >= 'a' && x <= 'z') temp = x - 'a' + 36;  // 36 以后
    return temp;
}

// 插入一个字符串
void insert(string str)
{
    int l = str.length();
    int p = 0;//从根节点往下找看是否有该字母
    for (int i = 0; i < l; i ++ )
    {
        int u = translate(str[i]);
        if (!son[p][u]) son[p][u] = ++ idx;
        p = son[p][u];
        cnt[p] ++ ;
    }
}

// 查询字符串出现的次数
int query(string str)
{
    int l = str.length();
    int p = 0;
    for (int i = 0; i < l; i ++ )
    {
        int u = translate(str[i]);
        if (!son[p][u]) return 0;
        p = son[p][u];
    }
    return cnt[p];
}

int main()
{
    int T;
    cin >> T;
    for(int i = 1; i <= T; i ++)
    {
        for(int a = 0; a <= idx; a ++)
        {
            for(int b = 0; b < 70; b ++)
            {
                son[a][b] = 0;
            }
            cnt[a] = 0;
        }
        idx = 0;   //字典初始化
        
        int n,q;
        cin >> n >> q;
        for(int j = 1; j <= n; j ++)
        {
            string s;
            cin >> s;
            insert(s);
        }
        
        for(int j = 1; j <= q; j ++)
        {
            string s;
            cin >> s;
            cout << query(s);
            if(j != q) cout << endl;
        }
        
        if(i != T) cout << endl;
    }
    return 0;
}