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

hdu-1251(字典树)

时间:2018-08-19 11:10:06      阅读:162      评论:0      收藏:0      [点我收藏+]

标签:space   using   query   div   mem   size   clu   时间   oid   

字典树模板题。

ps:数组要开大,40w左右才行,不然疯狂re

代码:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
int root;
int tot=1;
int trie[400500][30];
int cnt[400500];
void init()//最后清空,节省时间
{
    memset(trie,0,sizeof(trie));
    memset(cnt,0,sizeof(cnt));
   tot=1;//RE有可能是这里的问题
}
void build_trie(char *s)
{
    int len=strlen(s);
    root=0;
    for(int i=0;i<len;i++)
    {
        int id=s[i]-‘a‘;
        if(!trie[root][id])
            trie[root][id]=++tot;
        root=trie[root][id];
        cnt[root]++;
    }
}
int query(char *s)
{
    int len=strlen(s);
    root=0;
    for(int i=0;i<len;i++)
    {
        int id=s[i]-‘a‘;
        if(!trie[root][id])
            return false;
        root=trie[root][id];
    }
    return cnt[root];
}
int main()
{
    char s[105];
    char t[105];
    int ans=0;
    init();
    while(gets(s))
    {
        int x=strlen(s);
        if(x==0)
            break;
        build_trie(s);
    }
    while(gets(t))
    {
        int flag=query(t);
        printf("%d\n",flag);
    }
}

  

hdu-1251(字典树)

标签:space   using   query   div   mem   size   clu   时间   oid   

原文地址:https://www.cnblogs.com/huangdao/p/9499501.html

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