字典树模板（HihoCoder - 1014）

最新推荐文章于 2021-04-17 18:08:14 发布

182cx

最新推荐文章于 2021-04-17 18:08:14 发布

阅读量141

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM模板文章标签：字典树模板

本文链接：https://blog.youkuaiyun.com/weixin_43871404/article/details/99546432

ACM模板专栏收录该内容

1 篇文章

订阅专栏

本文详细介绍了字典树（Trie）的数据结构及其在HihoCoder-1014问题中的应用。通过C++实现，讲解了如何使用字典树进行字符串的高效插入和搜索，特别关注于单词个数的统计功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

字典树模板（HihoCoder - 1014）

#include <bits/stdc++.h>
using namespace std;
struct node{
    int z[30];int sum=0;
}tr[1000010];				//数组大小开n(插入最多单词个数)*l(最长单词长度)
int num=0;
void tr_insert(char s[])
{
    int st=strlen(s);
    int rt=0;
    for(int i=0;i<st;i++){
        int id=s[i]-'a';
        if(!tr[rt].z[id]){
            tr[rt].z[id]=++num;
        }
        rt=tr[rt].z[id];
         tr[rt].sum++;				//标记前面字符相同，以s[i]结尾的单词个数
    }
}
int tr_search(char s[])
{
    int st=strlen(s);
    int rt=0;
    int cou=0;
    for(int i=0;i<st;i++){
        int id=s[i]-'a';
        if(!tr[rt].z[id]){cou++;break;}		//如果不存在该节点跳出循环，标记一下
        rt=tr[rt].z[id];
    }
    if(cou==0)return tr[rt].sum;
    else return 0;
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        char c[11];
        scanf("%s",c);
        tr_insert(c);
    }
    int m;
    scanf("%d",&m);
    int ans=0;
    for(int i=1;i<=m;i++){
        char c[11];
        scanf("%s",c);
        cout<<tr_search(c)<<endl;
    }
    return 0;

}