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[];
     int key;
     char ans_s[];
 }root;
 void insert(char *str,char *s)
 {
     int len=strlen(str);
     node *p=&root,*q;
     for(int i=;i<len;i++)
     {
         int id=str[i]-'a';
         if(p->next[id]==NULL)
         {
             q=(node *)malloc(sizeof(root));
             q->key=;
             for(int j=;j<;j++)
                 q->next[j]=NULL;
             p->next[id]=q;
             p=p->next[id];
         }
         else
             p=p->next[id];
     }
     if(p->key!=-)
     {
         strcpy(p->ans_s,s);
         p->key=-;
     }
 }
 void find(char *str)
 {
     int len=strlen(str);
     node *p=&root;
     for(int i=;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==-)
         printf("%s",p->ans_s);
     else
         printf("%s",str);
     return;

 }
 int main()
 {
     int i;
     char s1[],s2[],s[];
     for(i=;i<;i++)
         root.next[i]=NULL;
     scanf("%s",s1);
     while(scanf("%s",s1)!=EOF&&strcmp(s1,"END")!=)
     {
         scanf("%s",s2);
         insert(s2,s1);
     }
     scanf("%s",s1);
     getchar();
     while(gets(s)&&strcmp(s,"END")!=)
     {
         int k=,len=strlen(s);
         for(i=;i<len;i++)
         {
             while(s[i]>='a'&&s[i]<='z')//满足是字母，wa了一次
             {
                 s2[k++]=s[i++];
             }
             s2[k]='\0';
             k=;
             find(s2);
             printf("%c",s[i]);
         }
         printf("\n");
     }

     return ;
 }