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;
}