算法刷题笔记 Trie字符串统计（C++实现）

题目描述

• 维护一个字符串集合，支持两种操作：
  • I x 向集合中插入一个字符串x；
  • Q x 询问一个字符串在集合中出现了多少次。
• 共有N个操作，所有输入的字符串总长度不超过10^5，字符串仅包含小写英文字母。

输入格式

• 第一行包含整数N，表示操作数。
• 接下来N行，每行包含一个操作指令，指令为I x或Q x中的一种。

输出格式

• 对于每个询问指令Q x，都要输出一个整数作为结果，表示x在集合中出现的次数。
• 每个结果占一行。

数据范围

• 1 ≤ N ≤ 2∗10^4

基本思路

• 本题是Trie树的模板题。Trie树又被称为单词查找树或字典树，其最大的作用就是快速查找和存储字符串，本题就是其在字符串存储和查找方面的应用。
• Trie树是一种特殊的树结构，其主要的操作是插入字符串。从字符串的第一个字符开始，每一个字符作为一个结点，该字符的下一个字符作为该结点的下一个结点（如果结点还不存在则创建）。每一个结点都可以记录以该结点结尾的字符串的个数，可以方便地进行查找。

实现代码

#include <iostream>
using namespace std;

const int N = 100010;

int next_node[N][26];  // 记录每一个字母结点的下一个字母结点的下标
int count[N];          // 记录以每一个结点结尾的字符串的个数
int index;             // 记录当前正在创建的结点的下标（根结点下标为0）

// 向trie树中插入新字符串的函数
void insert_to_trie(const string& x)
{
    // 记录当前遍历到的结点的下标，初始为编号为0的trie树根节点
    int current_node_index = 0; 
    // 通过for循环遍历字符串的每一个字符
    for(int i = 0; i < x.length(); ++ i)
    {
        // 记录当前的字符，并将其转换为数字
        int current_char = x[i] - 'a';
        // 通过当前遍历到的结点的son数组，判定是否存在以当前字符结尾的字符串，如果没有则创建一个并记录
        if(next_node[current_node_index][current_char] == 0) next_node[current_node_index][current_char] = ++ index;
        // 指向下一个结点的下标
        current_node_index = next_node[current_node_index][current_char];
    }
    // 遍历完成后，根据当前结点的下标修改字符串的出现次数
    count[current_node_index] ++;
}

// 在trie树种查找新字符串的函数
int query_trie(const string& x)
{
    // 记录当前遍历到的结点的下标，初始为编号为0的trie树根节点
    int current_node_index = 0; 
    // 通过for循环遍历字符串的每一个字符
    for(int i = 0; i < x.length(); ++ i)
    {
        // 记录当前的字符，并将其转换为数字
        int current_char = x[i] - 'a';
        // 通过当前遍历到的结点的son数组，判定是否存在以当前字符结尾的字符串，如果没有则创建一个并记录
        if(next_node[current_node_index][current_char] == 0) return 0;
        // 指向下一个结点的下标
        current_node_index = next_node[current_node_index][current_char];
    }
    // 遍历完成后，根据当前结点的下标修改字符串的出现次数
    return count[current_node_index];
}

int main(void)
{
    int n;
    cin >> n;
    for(int i = 0; i < n; ++ i)
    {
        char operation;
        string x;
        cin >> operation >> x;
        if(operation == 'I') insert_to_trie(x);
        else if(operation == 'Q') cout << query_trie(x) << endl;
    }
    return 0;
}