hdu 2072 单词数 字典树

题目链接

思路:

这个题其实好多方法可以做,map set 都可以搞过去.刚学字典树,所以用一下模板...

这个题只用到了插入，而且把字典树中的cnt稍微改动一下即可,那一个存的是到此位置时以该字符串为前缀的字符串的数量.这里表示的是该单词最后的尾节点在字符串中是否出现过,也就是是否出现过该单词.

#include   
#include   
#include   
#include   
#include   
using namespace std;  
const int maxn=1e6+10;
char s[maxn];  
char s2[maxn];  
int ch[maxn][26];  
int cnt[maxn], tmp, ans;
//这里就将cnt数组进行了简单的变化,cnt数组表示最后该单词的尾节点在字典树中是否出现过
//未出现过,单词总数就+1,否则不变.  
void insert(char *a)  
{  
    int rt = 0;  
    for (int i = 0; a[i]; i++)  
    {  
        int c = a[i] - 'a';  
        if (ch[rt][c] == 0)  
        {  
            cnt[tmp] = 0;   
            ch[rt][c] = tmp++;  
        }  
        rt = ch[rt][c];  
    }  
    if (cnt[rt] == 0) ans++;  
    cnt[rt] = 1;  
}  
  
int main()  
{  
    while (gets(s) && strcmp(s, "#"))  
    {  
        int i, j, top = 0;  
        tmp = 1, ans = 0;  
        cnt[0] = 0;  
        memset(ch[0], 0, sizeof(ch[0]));  
        for (i = 0; i <= strlen(s); i++)  
        {  
            if (s[i] >= 'a' && s[i] <= 'z')    
            s2[top++] = s[i];  
            else  
            {  
                if (top != 0)  
                {s2[top] = '\0';  
                top = 0;  
                insert(s2);}  
            }  
        }  
        printf("%d\n", ans);  
    }  
    return 0;  
}