835. Trie字符串统计

最新推荐文章于 2024-08-29 19:36:07 发布

蔓蔓的学习记录

最新推荐文章于 2024-08-29 19:36:07 发布

阅读量81

点赞数

分类专栏： AcWing 文章标签：算法 c++ 开发语言

本文链接：https://blog.csdn.net/manerzi/article/details/130068518

版权

AcWing 专栏收录该内容

40 篇文章 0 订阅

订阅专栏

该程序使用AC自动机数据结构，实现了对字符串的插入操作和查询频率的功能。它维护了一个儿子数组son和计数器cnt，用于存储字符串及其出现次数。在插入字符串时，为每个节点动态分配下标，并更新计数器；查询时，遍历字符串并返回对应节点的计数。

摘要由CSDN通过智能技术生成

在这里插入图片描述

AC代码：

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

const int N=1e5+10;

//字符串“总”长度不超过N
//son[i][j]存储下标为i，且指向字母为j的结点下标，如果为0，说明该结点不存在
int son[N][26],cnt[N],idx;//idx给每个结点分配下标，头结点下标为0，cnt为字符串的数量
char str[N];

void insert(){
    int h=0;
    for(int i=0;str[i];i++){
        int c=str[i]-'0';
        if(!son[h][c]) son[h][c]=++idx;//如果没有 就分配一个
        h=son[h][c];//下一个结点下标
    }
    cnt[h]++;
}

int query(){
    int h=0;
    for(int i=0;str[i];i++){
        int c=str[i]-'0';
        if(!son[h][c]) return 0;
        h=son[h][c];
    }
    return cnt[h];
}

int main(){
    int q;
    cin>>q;
    while(q--){
        char op;
        cin>>op>>str;
        if(op=='I') insert();
        else cout<<query()<<endl;
    }
    return 0;
}