ZOJ 1109 Language of FatMouse 【Trie树】

<题目链接>

题目大意：

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

解体分析：

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

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
 
const int MAXN = 100010;
const int N = 11;
const int M = 26;
 
struct Node{
    int index;
    Node* child[M];
    Node(){
        index = -1;
        for(int i = 0 ; 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 = 0 ; 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 = 0 ; i < len ; i++){
        int num = str[i]-'a';
        if(s->child[num] == NULL)
            return -1;
        s = s->child[num];
    }
    return s->index;
}
 
int main(){
    pos = 0;
    root = new Node();
 
    char str[N*3];
    char tmp[N];
    int cnt = 0;
    
    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 == -1)
            printf("eh
");
        else
            printf("%s
" , word[index]);
    }
    return 0;
}

2018-10-29