Trie树 [Hihocoder]

来源：互联网发布：mysql数据库视频教程编辑：程序博客网时间：2024/05/22 12:32

输入
输入的第一行为一个正整数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给出的字符串为前缀的单词的个数。

样例输入

5babaabbabbbaaaaabbaaaaaabaababaababb5babbbaabaaababbbbbabbaab

样例输出
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;}

0 0