ZOJ 1109 Language of FatMouse 【Trie树】

<题目链接>

题目大意：

刚开始每行输入两个单词，第二个单词存入单词库，并且每行第二个单词映射着对应的第一个单词。然后每行输入一个单词，如果单词库中有相同的单词，则输出它对应的那个单词，否则输出“eh”。

解体分析：

本题是trie树的模板题，在建树的时候，在每个单词的结尾节点储存对应单词的序号即可。当然，本题用map也可做，但是效率不够高。

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 using namespace std;

 const int MAXN = ;
 const int N = ;
 const int M = ;

 struct Node{
     int index;
     Node* child[M];
     Node(){
         index = -;
         for(int i =  ; i < M ; i++)
            child[i] = NULL;
     }
 };
 Node* root;

 int pos;
 char word[MAXN][N];

 void insert(int index , char* str){
     Node* s = root;
     int len = strlen(str);
     for(int i =  ; i < len ; i++){
         int num = str[i]-'a';
         if(s->child[num] == NULL)
             s->child[num] = new Node();
         s = s->child[num];
     }
     s->index = index;   //该单词的结束节点记录所对应字符串的序号
 }

 int search(char *str){
     Node* s = root;
     int len = strlen(str);
     for(int i =  ; i < len ; i++){
         int num = str[i]-'a';
         if(s->child[num] == NULL)
             return -;
         s = s->child[num];
     }
     return s->index;
 }

 int main(){
     pos = ;
     root = new Node();

     char str[N*];
     char tmp[N];
     int cnt = ;

     while(gets(str) && str[] != '\0'){
         sscanf(str , "%s %s" , word[cnt] , tmp);
         insert(cnt++ , tmp);
     }

     while(scanf("%s" , tmp) != EOF){
         int index = search(tmp);
         if(index == -)
             printf("eh\n");
         else
             printf("%s\n" , word[index]);
     }
     return ;
 }

2018-10-29