hdu1251 字典树trie 模板题

//字典树模板题.题意:给一个库,每次查询,求以之为前缀的单词数量。
#include<iostream>
#include<string>
#include<vector>
#include<cstdio>
#include<cstring>
using namespace std;
int tree[500000][27];
int w[500000];                  //记录结点属性。
int numv=0;               //  全局变量,记录顶点编号     
int n;
void insert(string s)        //建树  tree[u][i]:结点u的第i个孩子('a'为0,以此类推).
{                           //值是i对应的孩子结点编号.这个要遍历孩子边,要遍历26次,
                           //否则添加vector<vector<> >,来保存边也可.
    int u=0;
    int len=s.size();
    for(int i=0;i<len;i++)
    {
        if(tree[u][s[i]-'a']==0)    //新建顶点
        {
            tree[u][s[i]-'a']=++numv;
        }
        u=tree[u][s[i]-'a'];     //往下找
        w[u]++;                 
    }
}
int find(string s)
{
    int u=0;
    int len=s.size();
    for(int i=0;i<len;i++)      //找不到
    {
        if(tree[u][s[i]-'a']==0)
           return 0;
        u=tree[u][s[i]-'a'];
    }
    return w[u];

}
int main()
{
    string s;
    while(1)          //一直读入字符串,到换行为止
    {
        cin>>s;
        insert(s);
        getchar();
        if(cin.peek()=='\n')break;
    }
    while(cin>>s)
    {
        cout<<find(s)<<endl;
    }
    return 0;
}

    原文作者:Trie树
    原文地址: https://blog.csdn.net/u011498819/article/details/22102187
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞