POJ 2503 Trie

链接：

http://poj.org/problem?id=2503

题意：

给定一些字符串以及它在外星语言中的对应翻译，现在有若干外星语言中的串，要把它们翻译成英语

题解：

这道题map，hash，trie都是可以做的

然而我用g++提交发现map和trie都超时了，换成c++就全都过了

map用了1141ms，trie只要485ms

代码：

vector<string> word;
int pi =1;

struct Node {
    int next[26];
    int value;
    bool end;
}tree[MAXN * 10];

void insert(string keyword, int value) {
    int index, p, i;
    for (i = p = 0; keyword[i]; i++) {
        index = keyword[i] - 'a';
        if (tree[p].next[index] == 0)
            tree[p].next[index] = pi++;
        p = tree[p].next[index];
    }
    tree[p].value = value;
    tree[p].end = 1;
}

int query(string keyword) {
    int index, p, i;
    for (i = p = 0; keyword[i]; i++) {
        index = keyword[i] - 'a';
        if (tree[p].next[index] == 0) return 0;
        p = tree[p].next[index];
    }
    if (tree[p].end) return tree[p].value;
    return 0;
}

int main() {
    string s, a, b;
    word.pb("eh");
    while (getline(cin, s) && s != "") {
        int fg = 1;
        rep(i, 0, s.length()) {
            if (s[i] == ' ') fg = 0;
            else if (fg) a += s[i];
            else b += s[i];
        }
        word.pb(a);
        insert(b, word.size() - 1);
        a = b = "";
    }
    while (cin >> a) cout << word[query(a)] << endl;
    return 0;
}