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

hdu-2072(字典树)

时间:2018-08-19 20:05:42      阅读:151      评论:0      收藏:0      [点我收藏+]

标签:pre   cstring   i++   set   char s   amp   ios   ems   init   

字典树模板题

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
int tot=1;
int root;
int trie[100500][30];
int flagg[100500];
int visit[100500];
char s[100500];
char t[10050];
int cnt;
void init()
{
    memset(flagg,0,sizeof(flagg));
    memset(trie,0,sizeof(trie));
    memset(visit,0,sizeof(visit));
    tot=1;cnt=0;
}
void build_trie(char *x)
{
    int len=cnt;
    root=0;
    for(int i=0;i<len;i++)
    {
        int id=x[i]-‘a‘+1;
        if(!trie[root][id])
            trie[root][id]=tot++;
        root=trie[root][id];
    }
    flagg[root]=1;
}
bool query(char *x)
{
    int len=cnt;
    root=0;
    for(int i=0;i<len;i++)
    {
        int id=x[i]-‘a‘+1;
        if(!trie[root][id])
        {
            return false;
        }
        root=trie[root][id];
    }
    if(flagg[root]&&visit[root]==0)
    {
        visit[root]=1;
        return true;
    }
    return false;
}
int main()
{
    while(gets(s))
    {
        if(s[0]==‘#‘)
            break;
        init();
        int slen=strlen(s);
        if(s[slen-1]!=‘ ‘)
            s[slen]=‘ ‘;
        for(int i=0;i<=slen;i++)
        {
            if(s[i]==‘ ‘&&cnt!=0)
            {
                build_trie(t);cnt=0;
            }
            else if(s[i]!=‘ ‘)
            {
                t[cnt++]=s[i];
            }
        }
        int ans=0;
        for(int i=0;i<=slen;i++)
        {
            if(s[i]==‘ ‘&&cnt!=0)
            {
                int flag=query(t);cnt=0;
                if(flag)
                ans++;
            }
            else if(s[i]!=‘ ‘)
            {
                t[cnt++]=s[i];
            }
        }
        printf("%d\n",ans);
    }
}

  

hdu-2072(字典树)

标签:pre   cstring   i++   set   char s   amp   ios   ems   init   

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

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