【HDU 2072 单词数】字典树/SET

最新推荐文章于 2020-01-25 15:22:48 发布

原创最新推荐文章于 2020-01-25 15:22:48 发布 · 244 阅读

0 ·

CC 4.0 BY-SA版权

HDU 同时被 2 个专栏收录

33 篇文章

订阅专栏

字典树

17 篇文章

订阅专栏

本文介绍了一种使用Trie树解决HDU2072问题的方法，通过记录不同单词出现次数，利用Trie树进行高效查找与统计。

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

HDU2072
题意就是出现的不同单词个数
直接把字符全部插入Trie树中，然后统计所有具有flagg标记的节点个数就好了。
也可以边插入边统计，如果当前字符串结尾下标已经被标记，就不对答案做贡献，否则ans++.
HDU 2072 代码

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<sstream>
using namespace std;
const int maxn =2e6+5;
int tree[maxn][30];
bool flagg[maxn];
int tot;
void insert_(string str)
{
   int len=str.size();
   int root=0;
   for(int i=0;i<len;i++)
   {
       int id=str[i]-'a';
       if(!tree[root][id]) tree[root][id]=++tot;
       root=tree[root][id];
   }
   flagg[root]=true;
}
bool find_(string str)
{
    int len=str.size();
    int root=0;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(!tree[root][id]) return true;//该单词没有出现过
        root=tree[root][id];
    }
    if(flagg[root]) return false;//出现过，不再算贡献
    else return true;
}
string str1,str2;
int main()
{
    ios::sync_with_stdio(false);
    while(getline(cin,str1))
    {
        if(str1=="#") break;
        int ans=0;
        stringstream ss(str1);
        while(ss>>str2)
        {
           if(find_(str2))
           {
               ans++;
               insert_(str2);
           }
        }
        cout<<ans<<endl;
        for(int i=0;i<tot;i++)
        {
            flagg[i]=false;
            for(int j=0;j<30;j++)
                tree[i][j]=0;
        }
        tot=0;
    }
    return 0;
}