HDU 1247 Hat’s Words（字典树）

http://acm.hdu.edu.cn/showproblem.php?pid=1247

题意：

给出一些单词，问哪些单词可以正好由其他的两个单词首尾相连而成。

思路：

先将所有单独插入字典树，然后对于每一个单词，枚举它的分割点，去字典树中查找是否具有这两个单词。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 50000+5;

char s[maxn][105];
int num = 0;

struct Trie
{
    int son[26];
    int ends;
}t[maxn*100];

void init(int x)
{
    t[x].ends = 0;
    memset(t[x].son,0,sizeof(t[x].son));
}

void insert(char* s)
{
    int u = 0, n = strlen(s);
    for(int i=0;i<n;i++)
    {
        int c = s[i]-'a';
        if(!t[u].son[c])
        {
            num++;
            init(num);
            t[u].son[c] = num;
        }
        u = t[u].son[c];
    }
    t[u].ends = 1;
}

bool query(char* s)
{
    int u = 0, n = strlen(s);
    for(int i=0;i<n;i++)
    {
        int c = s[i]-'a';
        if(!t[u].son[c])  return false;
        u = t[u].son[c];
    }
    if(t[u].ends == 1)  return true;
    return false;
}

int main()
{
    //freopen("in.txt","r",stdin);
    int tot = 0;
    char s1[105],s2[105];
    while(~scanf("%s",s[tot++])) insert(s[tot-1]);
    for(int i=0;i<tot;i++)
    {
        int len = strlen(s[i]);
        for(int j=1;j<len;j++)
        {
            memset(s1,0,sizeof(s1));
            memset(s2,0,sizeof(s2));
            strncpy(s1,s[i],j);
            strncpy(s2,s[i]+j,len-j);
            if(query(s1) && query(s2)) {printf("%s
",s[i]);break;}
        }
    }
    return 0;
}