Hdu 1247 Hat's word

大意：给你一段单词，若给定的单词中有一个是另外两个单词合并而成的话，则输出。

 

思路：Trie+枚举，每输入一个单词，插入进Trie中，然后枚举它所有的前缀数，然后通过函数strncpy函数确定剩余的字母组成的单词，然后通过Trie的基本查找来确定是否包含两个单词。

 

CODE：

 

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

struct Trie
{
    Trie *next[26];
    int value;
    Trie()
    {
        for(int i = 0; i < 26; i++)
            next[i] = 0;
        value = 0;
    }
};

char save[50001][1001];

void insert(Trie *&root, char *s)
{
    int i = 0, branch = 0;
    Trie *p = root;
    if(!p)
    {
        p = new Trie();
        root = p;
    }
    while(s[i])
    {
        branch = s[i]-'a';
        if(!p->next[branch]) p->next[branch] = new Trie();
        p = p->next[branch];
        i++;
    }
    p->value = 1;
}

int find(Trie *root, char *s)
{
    int i = 0, branch = 0;
    Trie *p = root;
    if(!p) return 0;
    while(s[i])
    {
        branch = s[i]-'a';
        if(!p->next[branch]) return 0;
        p = p->next[branch];
        i++;
    }
    if(p->value) return 1;
    return 0;
}

int main()
{
    int tot = 0;
    Trie *root = NULL;
    while(~scanf("%s", save[tot]))
    {
        insert(root, save[tot]);
        tot++;
    }
    for(int i = 0; i < tot; i++)
    {
        for(int j = 0; j < strlen(save[i]); j++)
        {
            char temp1[101] = {'\0'};
            char temp2[101] = {'\0'};        //strncpy用法的特殊性
            strncpy(temp1, save[i], j);
            strncpy(temp2, save[i]+j, strlen(save[i])-j); //求后半部分单词
            if(find(root, temp1) && find(root, temp2))
            {
                printf("%s\n", save[i]);
                break;      //可能有重复的，所以必须退出。
            }
        }
    }
    return 0;
}