2390: 查单词--Power OJ（字典树）

最新推荐文章于 2021-09-24 00:51:34 发布

Wang_rush

最新推荐文章于 2021-09-24 00:51:34 发布

阅读量422

点赞数 1

CC 4.0 BY-SA版权

分类专栏：字典树 Power oj日记文章标签：字典树前缀树

本文链接：https://blog.youkuaiyun.com/weixin_44003265/article/details/97246259

Power oj日记同时被 2 个专栏收录

13 篇文章

订阅专栏

字典树

1 篇文章

订阅专栏

本文介绍了一种使用字典树（Trie）解决单词前缀查询问题的方法。通过构建字典树并统计各节点的子节点数量，可以高效地查询以特定字符串为前缀的单词数量。文章提供了详细的C++实现代码，包括插入单词和查询前缀两个主要功能。

Description
XX 最近遇到一个难题：老师交给他一本字典，这本字典里有100000 个单词(单词只有小写字母组成,并且不会有重复的单词出现),现在老师要他在这本字典中查找出以某个字符串为前缀的单词的数量(单词本身也是自己的前缀)。
Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是字典里的单词,一个空行代表单词表的结束。。第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串（长度不超过10）。

注意:本题只有一组测试数据,处理到文件结束.
Output
对于每个提问,输出以该字符串为前缀的单词的数量.
Sample Input
Raw
banana
band
bee
absolute
acm

ba
b
band
abc
Sample Output
Raw
2
3
1
0
Hint
简单字典树。。
Tags
————————————————————————————————————————————————————————
解题思路：
就是一个简单的字典树，对每个节点统计下面的子节点个数，注意输入时一个空行结束。

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int MAX_NODE = 1000000 + 10;
const int CHARSET = 26;
int trie[MAX_NODE][CHARSET] = {0};
int sum[MAX_NODE] = {0};
int k = 1;

void insert_tree(char *str){//建树
    int len = strlen(str);
    int p = 0;
    for(int i=0; i<len; i++){
        int c = str[i] - 'a';
        if(!trie[p][c]){
            trie[p][c] = k;
            k++;
        }
        p = trie[p][c];
        sum[p]++;
    }
}

int search_tree(char *str){//查询
    int len = strlen(str);
    int p = 0;
    for(int i=0; i<len; i++){
        int c = str[i] - 'a';
        if(!trie[p][c]) return 0;
        p = trie[p][c];
    }
    return sum[p];
}

int main(){
    char str[20];
    while (gets(str) && strlen(str)) {
        insert_tree(str);
    }
    while (gets(str))
            printf("%d\n", search_tree(str));
    return 0;
}