Trie树 [Hihocoder]

输入
输入的第一行为一个正整数n,表示词典的大小,其后n行,每一行一个单词(不保证是英文单词,也有可能是火星文单词哦),单词由不超过10个的小写英文字母组成,可能存在相同的单词,此时应将其视作不同的单词。接下来的一行为一个正整数m,表示小Hi询问的次数,其后m行,每一行一个字符串,该字符串由不超过10个的小写英文字母组成,表示小Hi的一个询问。

在20%的数据中n, m<=10,词典的字母表大小<=2.
在60%的数据中n, m<=1000,词典的字母表大小<=5.
在100%的数据中n, m<=100000,词典的字母表大小<=26.

本题按通过的数据量排名哦~

输出
对于小Hi的每一个询问,输出一个整数Ans,表示词典中以小Hi给出的字符串为前缀的单词的个数。

样例输入

5
babaab
babbbaaaa
abba
aaaaabaa
babaababb
5
babb
baabaaa
bab
bb
bbabbaab

样例输出
1
0
3
0
0
代码如下:(包括Trie树的建立,插入、查找)

#include <stdio.h>
#include <iostream>
#include <stdlib.h>
using namespace std;

class node{
    public:
    node(){
        for(int i=0;i<26;i++)
        next[i]=NULL;
        count=1;
    }
    node* next[26];
    int count;
};
class tree{
    public:
    tree(){
        root=new node();
    }
    void insert(string &word);
    int search(string &word);
    private:
    node *root;
};
void tree::insert(string &word){
    int len=word.length();
    node *proot=root;
    for(int i=0;i<len;i++){
        if(proot->next[word[i]-'a']==NULL)
        proot->next[word[i]-'a']=new node();
        else proot->next[word[i]-'a']->count++;
        proot=proot->next[word[i]-'a'];
    }
}
int tree::search(string &word){
    int len=word.length();
    node *proot=root;
    for(int i=0;i<len;i++){
        if(proot->next[word[i]-'a']==NULL)
        return 0;
        proot=proot->next[word[i]-'a'];
    }
    return proot->count;
}
int main(){
    int n;
    string word;
    tree T;
    cin>>n;
    while(n--){
        cin>>word;
        T.insert(word);
    }
    cin>>n;
    while(n--){
        cin>>word;
        cout<<T.search(word)<<endl;
    }
    return 0;
}
    原文作者:Trie树
    原文地址: https://blog.csdn.net/zhouyusong_bupt/article/details/51198854
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞

发表评论

电子邮件地址不会被公开。 必填项已用*标注