HihoCoder第二週（字典樹）

題目

輸入的第一行爲一個正整數n，表示詞典的大小，其後n行，每一行一個單詞，單詞由不超過10個的小寫英文字母組成，可能存在相同的單詞，此時應將其視作不同的單詞。接下來的一行爲一個正整數m，表示詢問的次數，其後m行，每一行一個字符串，該字符串由不超過10個的小寫英文字母組成，表示一個詢問。對於每一個詢問，輸出一個整數Ans,表示詞典中以給出的字符串爲前綴的單詞的個數。

---

樣例輸入
5
babaab
babbbaaaa
abba
aaaaabaa
babaababb
5
babb
baabaaa
bab
bb
bbabbaab
樣例輸出
1
0
3
0
0

---

#include<iostream>
#include <string>
using namespace std;
typedef struct trie
{
    char c;
    int count;
    struct trie* next[26];
} Trie;

int main()
{
    Trie* root = new Trie;
    for (unsigned i(0); i < 26; ++i)
        root->next[i] = NULL;
    int n, m;
    string s;
    cin >> n;
    while (n--)
    {
        cin >> s;
        Trie* p = root;
        for (unsigned i(0); i < s.size(); ++i)
        {
            unsigned j = s[i] - 'a';
            if (p->next[j])
            {
                ++p->next[j]->count;
            }
            else
            {
                Trie* q = new Trie;
                q->c = s[i];
                q->count = 1;
                for (unsigned k(0); k < 26; ++k)
                    q->next[k] = NULL;
                p->next[j] = q;
            }
            p = p->next[j];
        }
    }
    cin >> m;
    while (m--)
    {
        cin >> s;
        bool flag = false;
        Trie* p = root;
        for (unsigned i(0); i < s.size(); ++i)
        {
            unsigned j = s[i] - 'a';
            if (!p->next[j])
            {
                flag = true;
                break;
            }
            p = p->next[j];
        }
        if (flag) cout << '0' << endl;
        else cout << p->count << endl;
    }
    return 0;
}