【字典树】HDU 1247 Hat’s Words

题意：给你一个单词表 输出能由另外两个单词组成的单词

先判断一个单词的前缀是否为单词表中的单词，再判断后缀是否为单词表中的单词；

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<limits.h>
#include<ctype.h>
#include<math.h>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;
#include<queue>
#include<stack>
#include<vector>
#include<deque>
#include<set>
#include<map>
//N为最大点数
#define M 150
#define IN freopen ("in.txt" , "r" , stdin);
#define OUT freopen ("out.txt" , "w" , stdout);
//M为最大边数
typedef long long LL;
const int MAXN = 2999;//点数的最大值
const int MAXM = 2222;//边数的最大值
const int INF = 0x3f3f3f3f;
const int maxnode = 1000000;
const int sigma_size =30;
struct trie
{
int ch[maxnode][sigma_size];
int val[maxnode];
int sz; // 结点总数
trie()
{
sz=1; //初始化
memset(ch[0],0,sizeof(ch[0]));
}
int idx(char c)
{
return c-'a';
}
//插入字符串s，附加信息为v。注意v必须！=0，因为0表示 本结点不是单词结点
void insert(char *s)
{
int u=0,n=strlen(s);
for(int i=0; i<n; i++)
{
int c=idx(s[i]);
if(!ch[u][c])//结点不存在
{
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=0; //附加信息为0
ch[u][c]=sz++;//新建结点
}
u=ch[u][c];//往下走
}
val[u]++;//字符串的最后一个字符的附加信息为v
}
int find(char *s)
{
int u=0,n=strlen(s);
for(int i=0; i<n; i++)
{
int c=idx(s[i]);
if(!ch[u][c])//结点不存在
{
return 0;
}
u=ch[u][c];//往下走
}
return val[u];
}
} tre;
int main()
{
//IN;
char s[30010][20];
int i=0;
while(scanf("%s",s[i++])!=EOF)
{
tre.insert(s[i-1]);
}
for(int j=0; j<i; j++)
{

int c=tre.idx(s[j][0]);
int ac=tre.ch[0][c];
for(int k=0; s[j][k]!='\0'; k++)
{
if(tre.val[ac])//若前缀是
{
if(tre.find(&s[j][k+1]))//判断后缀
{
printf("%s\n",s[j]);
break;
}
}
c=tre.idx(s[j][k+1]);
if(tre.ch[ac][c]==0)//若前缀不在树上则break
break;
ac=tre.ch[ac][c];
}
}
return 0;
}