【oiClass 1502】查单词（Trie字典树）

题目描述

全国英语四级考试就这样如期来了，可是小y依然没有做好充分准备。为了能够大学毕业，可怜的小y决定作弊。
小y费尽心机，在考试的时候夹带了一本字典进考场，但是现在的问题是，考试的时候可能有很多的单词要查，小y能不能来得及呢？

输入

第一行一个整数N，表示字典中一共有多少个单词（N<=10000）。
接下来每两行表示一个单词，其中：
第一行是一个长度<=100的字符串，表示这个单词，全部小写字母，单词不会重复。
第二行是一个整数，表示这个单词在字典中的页码。
接下来是一个整数M，表示要查的单词数(M<=10000)。
接下来M行，每行一个字符串，表示要查的单词，保证在字典中存在。

输出

M行，每行一个整数，表示第I个单词在字典中的页数。

Trie字典树模板题，把每个单词的页数存在单词在trie里最后一个字母的位置即可。

#include <cstring>
#include <cstdio>
 
int n,m,x,cnt;
char s[100];
 
struct node{
    int val,son[26];
}trie[1000001];
 
void insert(char s[],int k){
    int len=strlen(s),x=0;
    for(int i=0;i<len;++i){
        int v=s[i]-'a';
        if(!trie[x].son[v])
            trie[x].son[v]=++cnt;
        x=trie[x].son[v];
    }
    trie[x].val=k;
}
 
int query(char s[]){
    int len=strlen(s),x=0;
    for(int i=0;i<len;++i){
        int v=s[i]-'a';
        x=trie[x].son[v];
    }
    return trie[x].val;
}
 
int main(void){
    scanf("%d",&n);
    for(int i=1;i<=n;++i){
        scanf("%s%d",&s,&x);
        insert(s,x);
    }
    scanf("%d",&m);
    for(int i=1;i<=m;++i){
        scanf("%s",&s);
        printf("%d\n",query(s));
    }
}