hdu2072—单词数（字典树）

单词数

Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 66960    Accepted Submission(s): 16855

 

Problem Description

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

 

 

Input

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

 

 

Output

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

 

 

Sample Input

you are my friend #

 

 

Sample Output

4

 

 

Author

Lily

 

 

Source

浙江工业大学网络选拔赛

 

 

Recommend

linle   |   We have carefully selected several similar problems for you:  2074 2054 2052 2024 2058 

其实这道题直接用stl就能过。

但是输入我不会欸，还是现学的。

代码：

#include<bits/stdc++.h>
using namespace std;
int main()
{
    string s;
    while(getline(cin, s) && s != "#") {
        istringstream sin(s);
        set<string> words;
        string w;
        
        while(sin >> w)
            words.insert(w);

        cout << words.size() << endl;
    }

    return 0;
}

字典树写法：

不知道为什么我用memset清空就会mle。必须要用for直接写。

有知道的欢迎留言。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=2000000+10;
const int inf=0x3f3f3f3f;
int tree[maxn][30];
bool flagg[maxn];
///int sum[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];
       /// sum[root]++;
    }
    flagg[root]=true;
}
int 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 0;///该单词没有出现过
        root=tree[root][id];
    }
    if(flagg[root])
        return 1;///出现过，不再算贡献
    else
        return 0;///没有出现过
}
char ss[maxn];
void init(){
  ///  memset(sum,0,sizeof(sum));
    tot=0;
   /// memset(flagg,0,flagg);
}
int main()
{
    int cnt=0;
    string str1,str2;
    while(getline(cin,str1)){
        if(str1=="#")
            break;
        int ans=0;
        stringstream ss(str1);
        while(ss>>str2){
            if(find_(str2)==0)
            {
                cnt++;
                insert_(str2);
            }
        } printf("%d\n",cnt);
        for(int i=0;i<tot;i++){
            flagg[i]=0;
            for(int j=0;j<30;j++){
                tree[i][j]=0;
            }
        }
        
        /*memset(flagg,0,sizeof(flagg));
        memset(tree,0,sizeof(tree));
        */
        tot=0;
        cnt=0;
    }
    return 0;
}

之前那个疑问在健大佬的帮助下解决了。

memset（tree,0,sizeof(tree[0]*tot)）这样子就能A了。

sizeof（）里面，tree[0]就是30，tot就是行数（高）。嗯。