【POJ】1451 T9

DFS+字典树。

 #include <cstdio>
 #include <cstring>
 #include <cstdlib>

 typedef struct Trie {
     int v;
     Trie *next[];
 } Trie;

 Trie root;
 int arr[] = {,,,,,,,,,,};
 char buf[], tmp[];
 char map[][];
 int max[], len;

 void create(char str[], int x) {
     int i, j, id;
     Trie *p = &root, *q;

     for (i=; str[i]!='\0'; ++i) {
         id = str[i] - 'a';
         if (p->next[id] == NULL) {
             q = (Trie *)malloc(sizeof(Trie));
             q->v = x;
             for (j=; j<; ++j)
                 q->next[j] = NULL;
             p->next[id] = q;
         } else {
             p->next[id]->v += x;
         }
         p = p->next[id];
     }
 }

 void del(Trie *t) {
     int i;

     if (t == NULL)
         return ;
     for (i=; i<; ++i)
         del(t->next[i]);
     free(t);
 }

 void find(Trie *t, int ti, int step) {
     int beg = arr[buf[step]-''];
     int end = arr[buf[step]-''+];
     int i;

     for (i=beg; i<end; ++i) {
         if (t->next[i] == NULL)
             continue;
         tmp[ti] = i+'a';
         if (t->next[i]->v > max[step]) {
             max[step] = t->next[i]->v;
             tmp[ti+] = '\0';
             strcpy(map[step], tmp);
         }
         if (step < len-)
             find(t->next[i], ti+, step+);
     }
 }

 int main() {
     int case_n, n, x;
     int i, j, t;

     scanf("%d", &case_n);

     for (t=; t<=case_n; ++t) {
         for (i=; i<; ++i)
             root.next[i] = NULL;
         scanf("%d", &n);
         while (n--) {
             scanf("%s %d", buf, &x);
             create(buf, x);
         }
         scanf("%d", &n);
         printf("Scenario #%d:\n", t);
         while (n--) {
             scanf("%s", buf);
             len = strlen(buf);
             --len;
             memset(max, -, sizeof(max));
             find(&root, , );
             for (i=; i<len; ++i) {
                 if (max[i] == -)
                     break;
                 printf("%s\n", map[i]);
             }
             for (j=i; j<len; ++j)
                 printf("MANUALLY\n");
             printf("\n");
         }
         printf("\n");
         del(&root);
     }

     return ;
 }