acwing 自我学习笔记-Trie字符串统计 835

#include <iostream>
#include <string>
#include <stdio.h>

using namespace std;

const int  N = 1e5;
int son[N][26];
int cnt[N];
int idx;//类似于struct结构中new一个新的结点的方式

void Insert(string x)
{
    int p = 0;//表明从根结点开始进行查找，同时根结点为空结点
    for (int i = 0; i < x.size(); i++)
    {
        int u = x[i] - 'a';//题目中字符串仅包含小写字母，故可用x[i]-'a'转换成数字存储
        if (!son[p][u])//如果不存在该结点则创建新的结点
        {
            idx++;//表明创建一个新的结点
            son[p][u] = idx;
        }
        p = son[p][u];//指向下一个结点
    }
    cnt[p]++;
}

int Query(string x)
{
    int p = 0;//从根结点开始查找
    for (int i = 0; i < x.size(); i++)
    {
        int u = x[i] - 'a';
        if (!son[p][u])
        {
            return 0;
        }
        else
        {
            p = son[p][u];
        }
    }
    return cnt[p];
}

int main()
{
    int n;
    scanf_s("%d",&n);
    string op, x;
    for (int i = 0; i < n; i++)
    {
        cin >> op >> x;
        if ("Q" == op)
        {
            cout << Query(x) << endl;
        }
        else
        {
            Insert(x);
        }
    }
    return 0;
}