hdu2072 单词数（字典树）

最新推荐文章于 2019-04-03 20:32:38 发布

原创最新推荐文章于 2019-04-03 20:32:38 发布 · 488 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#hdu

hdu 同时被 2 个专栏收录

231 篇文章

订阅专栏

数据结构-各种树

14 篇文章

订阅专栏

本文分享了一道使用Trie树解决的编程题实战经验，重点介绍了如何正确统计不同单词的数量，包括处理输入中可能存在的空白字符，并给出了完整的C++实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这题因为输入WA了N次，全是泪。

注意字符串前面有空格的情况，输入方法还是看的别人的。

最重要的一点，这题只是通过统计某单词是否出现来统计不同单词字数，切记要把p -> sum ++;放在外面，放在里面就不行。理论上来说也可以啊，最底层的字母sum都是0，可是就是WA，实在没辙，放for外面好了。

#include <stdio.h>
#include <algorithm>
#include <stdlib.h>
#include <string.h>
#include <iostream>

using namespace std;

typedef long long LL;

const int N = 1000005;
const int INF = 1e8;

struct Trie
{
    int sum;
    Trie *next[26];
    Trie()
    {
        sum = 0;
        for(int i = 0; i < 26; i ++)
            next[i] = 0;
    }
}*root;

int ans;

void inserttrie(Trie *p, char *str)
{
    int k = 0;
    while(str[k] != '\0')
    {
        int id = str[k] - 'a';
        if(p -> next[id] == 0)
        {
            p -> next[id] = new Trie;
        }
        p = p -> next[id];
        k ++;
    }
    if(p -> sum == 0) ans ++;
    p -> sum ++;//该单词是否出现过，注意放在外面
}

void Free(Trie *p)
{
    for(int i = 0; i < 26; i ++)
        if(p -> next[i]) Free(p -> next[i]);
    if(p) delete p;
}

int main()
{
   // freopen("in.txt", "r", stdin);
    char s[N], s0[1005];
    while(gets(s))
    {
        ans = 0;
        root = new Trie;
        if(s[0] == '#') break;
        int len = strlen(s);
        int j = 0;
        for(int i = 0; i < len; i ++)
        {
            while(s[i ++] == ' '); i --;
            if(i >= len) break;
            int j = 0;
            while(s[i] != ' ' && i < len)
                s0[j ++] = s[i ++];
            s0[j] = '\0';
            inserttrie(root, s0);
        }
        printf("%d\n", ans);
        Free(root);
    }
    return 0;
}