（字典树）HDU

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=1251

---

题意：先给定一些单词构成字典，然后再查一些在字典中以一些字母串为前缀的单词有几个。

---

分析：很基础很普通的字典树题目，但是用动态的字典树会超时，所以要用静态的字典树。

---

代码：

#include <set>
#include <map>
#include <list>
#include <cmath>
#include <queue>
#include <vector>
#include <bitset>
#include <string>
#include <cctype>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
#define inf (0x3f3f3f3f)
#define lnf (0x3f3f3f3f3f3f3f3f)
#define eps 1e-8
int sgn(double a){return a < -eps ? -1 : a < eps ? 0 : 1;}

//--------------------------

struct  Trie_node {
    int cnt;
    int next[26];
}tree[400010];
int nxt;

int  createTrieNode() {
    memset(&tree[nxt], 0, sizeof(Trie_node));
    return nxt++;
}

void Trie_insert(string& word) {
    int rt=0;
    int len=word.length();
    for(int i=0;i<len;i++){
        int id = word[i] - 'a';
        if(!tree[rt].next[id]){
            tree[rt].next[id]=createTrieNode();
        }
        rt=tree[rt].next[id];
        tree[rt].cnt++;
    }
}

int Trie_search(string& word) {
    int rt=0;
    int len=word.length();
    for(int i=0;i<len;i++){
        int id=word[i]-'a';
        if(!tree[rt].next[id])return 0;
        rt=tree[rt].next[id];
    }
    return tree[rt].cnt;
}

string word;

void init(){

}

void solve() {
    nxt=1;
    memset(&tree[0],0,sizeof(Trie_node));
    while(getline(cin,word)) {
        if(word=="")break;
        Trie_insert(word);
    }
    while(getline(cin,word)){
        cout<<Trie_search(word)<<endl;
    }
}

int main() {

#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
#endif
    iostream::sync_with_stdio(false);
    init();
    solve();
    return 0;
}