HDOJ 1075

字典树

9890974

2013-12-25 15:31:06

Accepted

1075

468MS

59832K

1342 B

G++

泽泽

 

 1 #include<stdio.h>
 2 #include<cstring>
 3 #include<cstdlib>
 4 struct node
 5 {
 6     node *next[26];
 7     int key;
 8     char ans_s[10];
 9 }root;
10 void insert(char *str,char *s)
11 {
12     int len=strlen(str);
13     node *p=&root,*q;
14     for(int i=0;i<len;i++)
15     {
16         int id=str[i]-'a';
17         if(p->next[id]==NULL)
18         {
19             q=(node *)malloc(sizeof(root));
20             q->key=1;
21             for(int j=0;j<26;j++)
22                 q->next[j]=NULL;
23             p->next[id]=q;
24             p=p->next[id];
25         }
26         else
27             p=p->next[id];
28     }
29     if(p->key!=-1)
30     {
31         strcpy(p->ans_s,s);
32         p->key=-1;
33     }
34 }
35 void find(char *str)
36 {
37     int len=strlen(str);
38     node *p=&root;
39     for(int i=0;i<len;i++)
40     {
41         int id=str[i]-'a';
42         if(p->next[id]==NULL)
43         {
44             printf("%s",str);
45             return ;
46         }
47         else
48         {
49             p=p->next[id];
50         }
51     }
52     if(p->key==-1)
53         printf("%s",p->ans_s);
54     else
55         printf("%s",str);
56     return;
57     
58 }
59 int main()
60 {
61     int i;
62     char s1[11],s2[11],s[3001];
63     for(i=0;i<26;i++)
64         root.next[i]=NULL;
65     scanf("%s",s1);
66     while(scanf("%s",s1)!=EOF&&strcmp(s1,"END")!=0)
67     {
68         scanf("%s",s2);
69         insert(s2,s1);
70     }
71     scanf("%s",s1);
72     getchar();
73     while(gets(s)&&strcmp(s,"END")!=0)
74     {
75         int k=0,len=strlen(s); 
76         for(i=0;i<len;i++)
77         {
78             while(s[i]>='a'&&s[i]<='z')//满足是字母，wa了一次
79             {
80                 s2[k++]=s[i++];
81             }
82             s2[k]='\0';
83             k=0;
84             find(s2);
85             printf("%c",s[i]);
86         }
87         printf("\n");
88     }
89     
90     return 0;
91 }

View Code