hdu1247(Hat’s Words)
我以为像a、aa这样的输入应该是没有输出的,结果还是要输出aa。
建树的时候就是常规建树,不过查找的时候要做一些变形:对于一个单词,从第一位检查有没有单词是它的前缀,如果有的话,再去检查它的后半部分是不是一个独立的单词,要满足这两次查找才能输出。
题意:给一些单词(以字典序输入),找出那些可以分成另外的两个单词的单词,以字典序输出;
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
using namespace std;
typedef struct Node
{
int flag;
struct Node *next[26];
}Node,*Tree;
char a[50001][30];
void Creat(Tree &T)
{
T=(Node *)malloc(sizeof(Node));
T->flag=0;
for(int i=0;i<26;i++)
{
T->next[i]=NULL;
}
}
void insert(Tree &T,char *s)
{
Tree p=T;
int t;
int l=strlen(s);
for(int i=0;i<l;i++)
{
t=s[i]-'a';
if(p->next[t]==NULL)
{
Creat(p->next[t]);
}
p=p->next[t];
}
p->flag=1;
}
int search(Tree T,char *s)
{
int t;
Tree p=T;
int l=strlen(s);
for(int i=0;i<l;i++)
{
t=s[i]-'a';
if(p->next[t]==NULL)
return 0;
p=p->next[t];
}
if(p->flag) return 1;
else return 0;
}
void D(Tree p)
{
for(int i=0;i<26;i++)
{
if(p->next[i]!=NULL)
D(p->next[i]);
}
free(p);
}
int main()
{
char s1[30];
char s2[30];
Tree T;
int kk=0;
Creat(T);
while(gets(a[kk])&&strlen(a[kk]))
{
insert(T,a[kk]);
kk++;
}
int i,j;
for(i=0;i<kk;i++)
{
int l=strlen(a[i]);
for(j=0;j<l;j++)
{
memset(s1,'\0',sizeof(s1));
memset(s2,'\0',sizeof(s2));
int kkk=0;
int jjj=0;
for(int kk=0;kk<=j;kk++)
s1[kkk++]=a[i][kk];
for(int jj=j+1;jj<l;jj++)
{
s2[jjj++]=a[i][jj];
}
int qq=search(T,s1);
int ww=search(T,s2);
if(qq&&ww)
{
printf("%s\n",a[i]);
break;
}
}
}
D(T);
return 0;
}