Hat’s Words（hdu 1247）（trie tree）

Hat’s Words（hdu 1247）

一个 “hat’s word”是一个单词可以恰好由字典中其他两个单词联接得到。给出你字典中的单词，你的工作是找出字典中所有的hat’s word。

输入：

每行为一个单词，由小写英文字母组成。所有单词按照字典顺序排列，总数不超过50,000。只有一组测试数据。

输出：

输出为所有的hat’s word，且按照字典顺序输出。

输入样例：

a 

ahat 

hat 

hatword 

hziee 

word

输出样例：

ahat 

hatword

分析：

将每个单词都插入到Trie树里，并按顺序依次判断每个单词是不是hat’s word。判断一个单词是不是hat’s word时，用暴力将这个单词分成前后两部分，判断这两部分是否都出现在Trie树里。

代码：

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

char ch[50050][100];

struct node
{
    struct node * next[26];
    bool value;
    node()
    {
        for(int i = 0; i < 26; i++)
            next[i] = NULL;
        value = false;
    }
}* root;

void insert(char s[])
{
    int k = 0;
    node *p = root;
    while(s[k] != '\0')
    {
        if( ! p->next[s[k]-'a'])
            p->next[s[k]-'a'] = new node();
        p = p->next[s[k]-'a'];
        k++;
    }
    p->value = true;
}

bool find(char s[])
{
    int k = 0;
    node *p = root;
    while(s[k] != '\0' && p->next[s[k]-'a'])
    {
        p = p->next[s[k]-'a'];
        k++;
    }
    if(s[k] == '\0' && p->value)
        return p->value;
    return false;
}

int main()
{
    int i = 0, k;
    root = new node();
    while(scanf("%s", ch[i]) != EOF)
    {
        insert(ch[i]);
        i++;
    }
    for(k = 0; k < i; k++)
    {
        int j;
        for(j = 1;j < strlen(ch[k]) - 1; j++)
        {
            char s1[100], s2[100];
            for(int jj = 0; jj < j; jj++)
                s1[jj] = ch[k][jj];
            s1[j] = '\0';
            strcpy(s2, ch[k] + j);
            if(find(s1) && find(s2))
            {
                cout << ch[k] << "\n";
                break;
            }
        }
    }
    return 0;
}