1004 Trie树

输入

输入的第一行为一个正整数n，表示词典的大小，其后n行，每一行一个单词（不保证是英文单词，也有可能是火星文单词哦），单词由不超过10个的小写英文字母组成，可能存在相同的单词，此时应将其视作不同的单词。接下来的一行为一个正整数m，表示小Hi询问的次数，其后m行，每一行一个字符串，该字符串由不超过10个的小写英文字母组成，表示小Hi的一个询问。

在20%的数据中n, m<=10，词典的字母表大小<=2.

在60%的数据中n, m<=1000，词典的字母表大小<=5.

在100%的数据中n, m<=100000，词典的字母表大小<=26.

本题按通过的数据量排名哦～

输出

对于小Hi的每一个询问，输出一个整数Ans,表示词典中以小Hi给出的字符串为前缀的单词的个数。

{优快云:CODE:

#include "stdafx.h"
#include <iostream>
#include <string>


using namespace std;


const int MAX = 26;


class TrieTree{
public:
struct TrieNode{
int cnt;
TrieNode * next[MAX];
};
TrieTree(){}
void InitTrieTree();
TrieNode * CreateNode();
void Insert(string);
int Search(string);
~TrieTree(){


}
private:
static int allocp;
TrieNode * head;
};


int TrieTree::allocp = -1;


void TrieTree::InitTrieTree(){
head = nullptr;
}


TrieTree::TrieNode * TrieTree::CreateNode(){
TrieNode * node = new TrieNode();
node->cnt = 0;
for (int i = 0; i < MAX; ++i){
node->next[i] = nullptr;
}
return node;
}


void TrieTree::Insert(string s){
if (head == nullptr)
head = CreateNode();
if (0 == s.size())
return;
TrieNode * p = head;
++p->cnt;
for (size_t i = 0; i < s.size(); ++i){
if (!p->next[s[i] - 'a']){
TrieNode * tmp = CreateNode();
p->next[s[i] - 'a'] = tmp;
}
p = p->next[s[i] - 'a'];
++p->cnt;
}
}


int TrieTree::Search(string s){
if (head == nullptr)
return 0;
if (0 == s.size())
return 0;
TrieNode *p = head;
for (size_t i = 0; i < s.size(); ++i){
if (s[i] - 'a' < 0 || s[i] - 'a' > MAX)
return 0;
if (p->next[s[i] - 'a']){
p = p->next[s[i] - 'a'];
}
else
return 0;
}
return p->cnt;
}
int _tmain(int argc, _TCHAR* argv[])
{
freopen("in.txt", "r", stdin);
TrieTree t;
TrieTree *pt = new TrieTree();
TrieTree *ptt = &t;
t.InitTrieTree();
int m, n;
string inp;
cin >> n;
for (int i = 0; i < n; ++i)
{
cin >> inp;
t.Insert(inp);
}
cin >> m;
while (m--){
cin >> inp;
cout << t.Search(inp) << endl;
}
return 0;
}}

思路：Trie树的概念比较简单，这道题比较坑的地方在于如果静态开了100000的大小（其实没问题！！！）提示Runtime Error其实是因为，他题目给的条件不符合的！会有数据超过100000的大小，所以之前开这么大根本不够！！！开成1000000就可以了，浪费了好多时间找RE的原因。。。