hdu-1247-Hat’s Words（字典树）

Hat’s Words(传送门)

题意：

在所给的单词表中，按字典序输出符合条件的 hat's word ，条件为： hat's word 恰好由所给单词表中的另外两个单词拼接而成。

思路：

对每一个单词判断，该单词所分为的两个单词是否都在所给单词表中，由于单词的数量最大为50000，判断拆分出的单词是否在单词表中是比较耗时的，所以为了提高效率，这里把所给单词另外建立成一棵字典树。

code：

#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
struct node    ///存储
{
    int cut;
    node *next[26];
    node(){
        cut=0;
        for(int i=0;i<26;i++){
            next[i]=NULL;
        }
    }
};
node *root;///根节点
char ss[50005][100];///存输入的 单词
void Insert(char *str)///插入到字典树中
{
    node *p=root;
    int len=strlen(str);
    for(int i=0;i<len;i++){
        int id=str[i]-'a';
        if(p->next[id]!=NULL){
            p=p->next[id];
        }
        else{
            p->next[id]=new node;
            p=p->next[id];
        }
    }
    p->cut=1;///单词的结束
}
bool LookUp(node *p,char *str)///查找后续
{
    //int len=strlen(str);
    for(int i=0;str[i]!='\0';i++){
        int id=str[i]-'a';
        if(p->next[id]==NULL) return false;
        p=p->next[id];
    }
    return p->cut;
}
void Find(node *p,int Size)
{
    for(int i=0;i<Size;i++){
        p=root;
        int len=strlen(ss[i]);
        for(int j=0;j<len;j++){
            if(p->cut==1){///找到前一个单词，在找后面有没有单词的结束。
                if(LookUp(root,&ss[i][j])){
                    puts(ss[i]);break;
                }
            }
            int id=ss[i][j]-'a';
            if(p->next[id]==NULL){
                break;
            }
            p=p->next[id];
        }
    }
}
int main()
{
    root=new node;
    int i;
    for(i=0;scanf("%s",ss[i])!=EOF;i++){
        Insert(ss[i]);
    }
    Find(root,i);
    return 0;
}