码迷,mamicode.com
首页 > 其他好文 > 详细

hdu1251(Trie树)

时间:2015-02-28 00:10:59      阅读:167      评论:0      收藏:0      [点我收藏+]

标签:

 

传送门:统计难题

分析:Trie树入门题,随便写写练下手感,统计每个节点被多少单词经过就可以了。

技术分享
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <queue>
#include <vector>
#define LL long long
#define N 500010
using namespace std;

char s[15];
struct Trie
{
    int cnt[N],child[N][27];
    int L,root;
    int newnode()
    {
        memset(child[L],0,sizeof(child[L]));
        return L++;
    }
    void init()
    {
        L=0;
        memset(cnt,0,sizeof(cnt));
        root=newnode();
    }
    void insert(char *s)
    {
        int now=root;
        for(int i=0;s[i];i++)
        {
            int id=s[i]-a;
            if(!child[now][id])
                child[now][id]=newnode();
            now=child[now][id];
            cnt[now]++;
        }
    }
    int query(char *s)
    {
        int now=root;
        for(int i=0;s[i];i++)
        {
            int id=s[i]-a;
            if(!child[now][id])return 0;
            now=child[now][id];
        }
        return cnt[now];
    }
}trie;
int main()
{
    trie.init();
    while(gets(s))
    {
        if(strlen(s)==0)break;
        trie.insert(s);
    }
    while(scanf("%s",s)==1)printf("%d\n",trie.query(s));
}
View Code

 

hdu1251(Trie树)

标签:

原文地址:http://www.cnblogs.com/lienus/p/4304518.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!