hdu 2072 单词数

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

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

int sum;       //不同单词的总数
char s[100000];
char st[500];

struct node
{
int n;  //用于标记从根到这里的单词出现的次数
node *next[26];
};

node root;

void tree(char *s, int len)
{
int i,j,id;
node *p,*q;
p = &root;
for(i = 0; i<len; i++)
{
id = s[i]-'a';
if(p->next[id]==NULL) //如果为空，则创建树，否则进入
{
q=(node *)malloc(sizeof(node));
q->n = 0;
for(j = 0; j<26; j++)
{
q->next[j] = NULL;
}
p->next[id] = q;
p=p->next[id];
}
else
{
p = p->next[id];
}
}
if(p->n==0)//如果这个单词没有出现过，则总数加一
{
sum++;
p->n++;
}
else
{
p->n++;
}
return ;
}

int main()
{
int n,i,j;
while(1)
{
sum = 0;
gets(s);
if(s[0]=='#')
{
break;
}
n = strlen(s);
for(i = 0; i < 26; i++)  //初始化树
{
root.n = 1;
root.next[i] = NULL;
}
j = 0;
for(i = 0; i < n; i++)
{
if(s[i]==' ') //用空格隔开，取出每个单词
{
tree(st,j);
j = 0;
}
else
{
st[j] = s[i];
j++;
}
}
/*if(s[n-1]!=' ')
{
tree(st,j);
}*/
tree(st,j);
printf("%d\n",sum);
}

return 0;
}