Trie字符串统计

题目描述 题目链接

 

 

题目分析

Trie字符树，可以高效存储字符串。

需要数据：
son[maxn][25] cnt[maxn] idx p

son[maxn][25] 表示结点的儿子 其本身的值代表结点编号 如son[1][0] 就代表1号结点的a儿子;

cnt[maxn] 表示以莫一个结点为结尾的单词出现的个数相当于标记此处有个单词，
如cnt[3]表示以3号结点的单词出现个数

idx 表示结点序号通过idx可以使每一个结点都有唯一的序号

p 代表存储字符串现在在那个结点

代码展示

#include<iostream>

using namespace std;

int const maxn=1e5+10;
int son[maxn][26],cnt[maxn],idx;

void insert(char str[])
{
    int p=0;                            //从头结点开始插入

    for(int i=0;str[i];i++)
    {
        int u=str[i]-'a';
        if(son[p][u]==0) son[p][u]=++idx;//如果son[p][u]为0表示这个结点没有str[i]这个儿子因此需要创造一个
        p=son[p][u];                     
    }
    cnt[p]++;
}

int query(char str[])
{

    int p=0;
    for(int i=0;str[i];i++)
    {
        int u=str[i]-'a';
        if(son[p][u]==0) return 0;
        p=son[p][u];
    }
    return cnt[p];
}

int main()
{
    int n; cin>>n;
    while(n--)
    {
        char aim,str[maxn]; cin>>aim>>str;
        if(aim=='I') insert(str);
        if(aim=='Q') cout<<query(str)<<endl;
    }
}