hdu1247Hat’s Words（trie树）

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

先找某个单词是否为一个单词的前缀 再用 这个单词减去前缀 判断剩下的部分是否为一个单词

#include <iostream>
#include<stdlib.h>
#include<cstdio>
#include<string.h>
using namespace std;
char c[50001][101];
struct node
{
    int flag;
    node *next[27];
    node()
    {
        flag = 0;
        memset(next,NULL,sizeof(next));
    }
};
void buildt(node *head,char *x)
{
    int i,j =0 ,k,d;
    k = strlen(x);
    node *p = head;
    for(i = 0 ; i < k ; i++)
    {
        d = x[i]-'a';
        if(p->next[d]==NULL)
        {
            p->next[d] = new node;
        }
        p = p->next[d];
    }
    p->flag = 1;
}
int judge(node *head,char *x)
{
    int i,d;
    node *p = head;
    while(*x!='\0')
    {
        d = *x-'a';
        if(p->next[d]==NULL)
        return 0;
        p = p->next[d];
        if(p->flag==1&&*(x+1)=='\0')//满足后一部分恰为一个单词
        return 1;
        x++;
    }
    return 0;
}
int search(node *head,char *x)
{
    int i,d,k = strlen(x),t = 0;
    node *p = head;
    for(; *x != '\0' ; )
    {
        d = *x-'a';
        if(p->next[d]==NULL)
            return 0;
        p = p->next[d];
        if(p->flag == 1&&judge(head,x+1))//找到一个作为它前缀的单词 查看后一部分是不是为另一个单词
        {
            return 1;
        }
        x++;
    }
    return 0;
}
int main()
{
    int i = 0,j,k,f;
    node *head = new node;
    while(gets(c[i])!=NULL)
    {
        buildt(head,c[i]);
        i++;
    }
    for(j =0 ; j < i ; j++)
    {
        if(search(head,c[j]))
        puts(c[j]);
    }
    return 0;
}