hdu 1247 Hat’s Words（字典树）

我以为像a、aa这样的输入应该是没有输出的，结果还是要输出aa。

建树的时候就是常规建树，不过查找的时候要做一些变形：对于一个单词，从第一位检查有没有单词是它的前缀，如果有的话，再去检查它的后半部分是不是一个独立的单词，要满足这两次查找才能输出。

老是从自己写的字典树里面闻出一种山寨的味道。

#include<stdio.h>#include<string.h>#include<malloc.h>#define N 50005char s[N][30];struct node{    node *a[26];    int flag;    int id;};node *root;void InsertTree(char *ss){    int i,ln;    node *cur,*s;    ln=strlen(ss);    cur=root;    for(i=0;i<ln;i++)    {        int temp;        temp=ss[i]-'a';        if(i==ln-1)        {            if(cur->a[temp]!=NULL)                cur->a[temp]->flag=1;            else            {                s=(node *)malloc(sizeof(node));                memset(s->a,0,sizeof(s->a));                s->flag=1;                s->id=-1;                cur->a[temp]=s;            }        }        else if(cur->a[temp]!=NULL)            cur=cur->a[temp];        else        {            s=(node *)malloc(sizeof(node));            memset(s->a,0,sizeof(s->a));            s->id=-1;            s->flag=0;            cur->a[temp]=s;            cur=s;        }    }    return ;}int FindTree(char *s,int flag,int k){    int i,ln;    node *cur;    cur=root;    ln=strlen(s);    if(!flag)    {        for(i=0;i<ln;i++)        {            int temp=s[i]-'a';            if(cur->a[temp]!=NULL)            {                if(cur->a[temp]->flag)                {                    cur->a[temp]->id=k;                    if(FindTree(s+i+1,1,k)==2)                        return 1;                }                cur=cur->a[temp];            }            else                return 0;        }    }    else    {        for(i=0;i<ln;i++)        {            int temp=s[i]-'a';            if(i==ln-1)            {                if(cur->a[temp]!=NULL/*&&cur->a[temp]->id!=k*/&&cur->a[temp]->flag)                    return 2;                else                    return 0;            }            else if(cur->a[temp]!=NULL)                cur=cur->a[temp];            else                return 0;        }    }    return flag;}int main(){    int k=0;    root=(node *)malloc(sizeof(node));    memset(root->a,0,sizeof(root->a));    root->flag=0;    root->id=-1;    while(gets(s[k]))    {        InsertTree(s[k]);        k++;    }    int i;    for(i=0;i<k;i++)    {        if(FindTree(s[i],0,i))            puts(s[i]);    }    return 0;}