hdu 2072 单词数

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

这个题我是用的字典树的方法来做的，首先一个字母一个字母判断，没有则创建，有则直接进入这一层。同时到单词结束时进行判断是否以前出现过。

AC代码：

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