首页 > 其他 > 详细

HDU 1251 统计难题

时间:2016-04-30 22:11:27      阅读:272      评论:0      收藏:0      [点我收藏+]

  字典树应用,每个节点上对应的cnt是以它为前缀的单词的数量

#include<stdio.h>
#include<string.h>
struct trie
{
    int cnt;
    trie *next[26];
};
trie *root=new trie;
void insert(char ch[])
{
    trie *p=root,*newnode;
    for(int i=0; ch[i]!=\0; i++)
    {
        if(p->next[ch[i]-a]==0)
        {
            newnode=new trie;
            for(int j=0; j!=26; j++)
            {
                newnode->next[j]=NULL;
            }
            newnode->cnt=1;
            p->next[ch[i]-a]=newnode;
            p=newnode;
        }
        else
        {
            p=p->next[ch[i]-a];
            p->cnt++;
        }
    }
}
int find(char ch[])
{
    trie *p=root;
    for(int i=0; ch[i]!=\0; i++)
    {
        if(p->next[ch[i]-a]!=NULL)
            p=p->next[ch[i]-a];
        else
            return 0;
    }
    return p->cnt;
}
int main()
{
    char ch[20];
    for(int i=0; i!=26; i++)
    {
        root->next[i]=NULL;
    }
    root->cnt=0;
    while(gets(ch)) //±ØÐëÓÃgets
    {
        if(!strcmp(ch,"")) break;
        insert(ch);
    }
    while(scanf("%s",ch)!=EOF)
    {
        printf("%d\n",find(ch));
    }
    return 0;
}

 

HDU 1251 统计难题

原文:http://www.cnblogs.com/jifahu/p/5449340.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!