tire树（字典树）

最新推荐文章于 2024-09-13 16:27:15 发布

kingo_

最新推荐文章于 2024-09-13 16:27:15 发布

阅读量117

点赞数 3

文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/2301_81235183/article/details/140360159

版权

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N=1008611;
int a,b,c,d;
int son[N][26],cnt[N],idx;
char str[N];
void insert(char str[])
{
    int p=0;
    for(int i=0;str[i];i++)
    {
        int u=str[i]-'a';
        if(!son[p][u])son[p][u]=++idx;
        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])return 0;
        p =son[p][u];
    }
    return cnt[p];
}
signed main()
{
    int n;
    cin>>n;
    while(n--)
    {
        char op[2];
        cin>>op>>str;
        if(op[0]=='I') insert(str);
        else  cout<<query(str);
        
    }
    return 0;
    
}