题目大意:
你要找到字典中所有two-word compound单词。一个two-word compound单词是字典中两个其他单词的连接
解题思路:
把每个单词拆成两部分,分别看字典中是否这两个单词
代码如下:
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<string.h>
#include<queue>
#include<limits.h>
#define MAX 120002
using namespace std;
typedef struct TRIE
{
int end;
TRIE *next[26];
}TRIE;
TRIE *head,*p;
void init()
{
int k;
p=(TRIE*)malloc(sizeof(TRIE));
for(k=0;k<26;k++)
p->next[k]=NULL;
p->end=0;
}
void Build(char *str)
{
int i,to,k;
int len=strlen(str);
head=p;
for(i=0;i<len;i++)
{
to=str[i]-'a';
if(head->next[to]==NULL)
{
head->next[to]=(TRIE*)malloc(sizeof(TRIE));
for(k=0;k<26;k++)
head->next[to]->next[k]=NULL;
head->next[to]->end=0;
}
head=head->next[to];
}
head->end=1;
}
int compound(char *str)
{
int len=strlen(str);
int k,i,too;
TRIE *tmp=p;
for(i=0;i<len-1;i++)
{
int to=str[i]-'a';
if(tmp->next[to]->end==1)
{
k=i+1;
head=p;
while(k!=len)
{
too=str[k]-'a';
if(head->next[too]!=NULL)
head=head->next[too];
else
break;
k++;
}
if(k==len&&head->end==1)
return 1;
}
tmp=tmp->next[to];
}
return 0;
}
char s[MAX][15];
int main()
{
int num=0,i;
init();
while(~scanf("%s",s[num]))
{
Build(s[num]);
num++;
}
for(i=0;i<num;i++)
{
if(strlen(s[i])>1)
{
if(compound(s[i])==1)
puts(s[i]);
}
}
return 0;
}