HDU 1247 Hat’s Words(字典树)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1247

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
char str1[16],str2[16],map[50005][16];
using namespace std;
struct node
{
    bool flag;
    node *next[26];
    node()
    {
        flag=false;
        for(int i=0;i<26;i++) next[i]=NULL;
    }
}*root;
void insert(char *s)
{
    node *p;
    p=root;
    int i,len;
    len=strlen(s);
    if(len==0)  return;
    for(i=0;i<len;i++)
    {
        if(p->next[s[i]-'a']!=NULL)
           p=p->next[s[i]-'a'];
        else
        {
            p->next[s[i]-'a']=new node();
            p=p->next[s[i]-'a'];
        }
    }
    p->flag=true;
}
bool search(char *s)
{
    node *p;
    p=root;
    int i,len=strlen(s);
    if(len==0)  return false;
    for(i=0;i<len;i++)
    {
        if(p->next[s[i]-'a']!=NULL)
           p=p->next[s[i]-'a'];
        else return false;
    }
    return p->flag;
}
int main()
{
   root=new node();
   int ni,nj,i,j,k,len;
   for(i=-1;scanf("%s",map[++i])!=EOF;insert(map[i]));
   for(j=0;j<=i;j++)//将单词分成两部分，放到树状数组里去找
   {
       len=strlen(map[j]);
       for(k=1;k<len;k++)
       {
           for(ni=0;ni<k;ni++)   str1[ni]=map[j][ni];
           for(nj=k;nj<len;nj++) str2[nj-k]=map[j][nj];
           str1[ni]=str2[nj-k]='\0';
           if(search(str1)&&search(str2))
           {
               printf("%s\n",map[j]);
               break;
           }
       }
   }
   return 0;
}