hdu2072 单词数（字典树）

这题因为输入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
, 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;
}