HDOJ 1075

字典树

9890974

2013-12-25 15:31:06

Accepted

1075

468MS

59832K

1342 B

G++

泽泽

#include<stdio.h>
#include<cstring>
#include<cstdlib>
struct node
{
    node *next[26];
    int key;
    char ans_s[10];
}root;
void insert(char *str,char *s)
{
    int len=strlen(str);
    node *p=&root,*q;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(p->next[id]==NULL)
        {
            q=(node *)malloc(sizeof(root));
            q->key=1;
            for(int j=0;j<26;j++)
                q->next[j]=NULL;
            p->next[id]=q;
            p=p->next[id];
        }
        else
            p=p->next[id];
    }
    if(p->key!=-1)
    {
        strcpy(p->ans_s,s);
        p->key=-1;
    }
}
void find(char *str)
{
    int len=strlen(str);
    node *p=&root;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(p->next[id]==NULL)
        {
            printf("%s",str);
            return ;
        }
        else
        {
            p=p->next[id];
        }
    }
    if(p->key==-1)
        printf("%s",p->ans_s);
    else
        printf("%s",str);
    return;
    
}
int main()
{
    int i;
    char s1[11],s2[11],s[3001];
    for(i=0;i<26;i++)
        root.next[i]=NULL;
    scanf("%s",s1);
    while(scanf("%s",s1)!=EOF&&strcmp(s1,"END")!=0)
    {
        scanf("%s",s2);
        insert(s2,s1);
    }
    scanf("%s",s1);
    getchar();
    while(gets(s)&&strcmp(s,"END")!=0)
    {
        int k=0,len=strlen(s); 
        for(i=0;i<len;i++)
        {
            while(s[i]>='a'&&s[i]<='z')//满足是字母，wa了一次
            {
                s2[k++]=s[i++];
            }
            s2[k]='';
            k=0;
            find(s2);
            printf("%c",s[i]);
        }
        printf("
");
    }
    
    return 0;
}