HDU 2072 字典树

http://acm.hdu.edu.cn/showproblem.php?pid=2072

 

Problem Description

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

Input

有多组数据，每组一行，每组就是一篇小文章。每篇小文章都是由小写字母和空格组成，没有标点符号，遇到#时表示输入结束。

Output

每组只输出一个整数，其单独成行，该整数代表一篇文章里不同单词的总数。

Sample Input

you are my friend #

Sample Output

4

Author

Lily

Source

浙江工业大学网络选拔赛

思路：map、set都可以做这道题，不过最近在学字典树，所以就用字典树写了。用vis标记一个单词是否已经出现过，最后统计即可。也可以在插入的同时统计。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<string>
#include<sstream>
#include<cmath>
#include<map>
#include<algorithm>
#define INF 0x3f3f3f3f
typedef long long ll;
using namespace std;

int tree[200000][26];
int vis[200000];
string s1,s;
int tot;

void Insert()
{
    int root=0;
    int len=s.size();
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'a';
        if(!tree[root][id])
            tree[root][id]=++tot;
        root=tree[root][id];
    }
    vis[root]=1;
}

int query()
{
    int cnt=0;
    for(int i=1;i<=tot;i++)
        if(vis[i])
            ++cnt;
    return cnt;
}

int main()
{
    while(getline(cin,s1)&&s1[0]!='#')
    {
        stringstream ss(s1);//向流ss中传值
        while(ss>>s)//向s中写入值
            Insert();
        printf("%d\n",query());
        memset(tree,0,sizeof(tree));
        memset(vis,0,sizeof(vis));
        tot=0;
    }
    return 0;
}