HDU - 2072——单词数

最新推荐文章于 2024-01-02 20:12:15 发布

creator平

最新推荐文章于 2024-01-02 20:12:15 发布

阅读量173

点赞数

分类专栏：字典树

本文链接：https://blog.csdn.net/weixin_41380961/article/details/80200061

版权

字典树专栏收录该内容

6 篇文章 0 订阅

订阅专栏

lily的好朋友xiaoou333最近很空，他想了一件没有什么意义的事情，就是统计一篇文章里不同单词的总数。下面你的任务是帮助xiaoou333解决这个问题。

Input 有多组数据，每组一行，每组就是一篇小文章。每篇小文章都是由小写字母和空格组成，没有标点符号，遇到#时表示输入结束。 Output 每组只输出一个整数，其单独成行，该整数代表一篇文章里不同单词的总数。 Sample Input

you are my friend
#

Sample Output

思路:简单的字典树模板，以下为使用指针构建的字典树。

#include<algorithm>
#include<string.h>
#include<stdio.h>
using namespace std;
char a[1100],b[110];
int ans;
struct Trie
{
    int v;
    Trie *next[26];
};
Trie root;
void create_Trie(char *str)
{

    Trie *p=&root,*q;
    while(*str)
    {
        int di=*str-'a';
        if(p->next[di]==NULL)
        {
            q=(Trie *)malloc(sizeof(Trie));
            for(int i=0; i<26; i++)
                q->next[i]=NULL;
            q->v=0;
            p->next[di]=q;
        }
        p=p->next[di];
        str++;
    }

    if(p->v==0)
    {
        ans++;
    }
    p->v++;
}
void clear_Trie(Trie *p)
{
    for(int i=0; i<26; i++)
    {
        if(p->next[i]!=NULL)
        {
            clear_Trie(p->next[i]);
            p->next[i]=NULL;
        }
    }
    if(p!=&root)//保留头节点
        free(p);
}
int main()
{
    ans=0;
    while(gets(a),a[0]!='#')
    {
        int len=strlen(a);
        int l=0;
        for(int i=0; i<=len; i++)
        {
            if((a[i]==' '||i==len)&&l>0)
            {
                b[l]='\0';
                create_Trie(b);//构建树
                l=0;
            }
            else if(a[i]!=' ')
                b[l++]=a[i];
        }
        printf("%d\n",ans);
        clear_Trie(&root);//释放树
        ans=0;
    }
}