hihocoder-num1014

I am a coder
Trie树
时间限制:10000ms
单点时限:1000ms
内存限制:256MB

描述

求相同特定前缀单词个数

输入

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

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

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

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

输出

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

样例输入

1
2
3
4
5
6
7
8
9
10
11
12
5
babaab
babbbaaaa
abba
aaaaabaa
babaababb
5
babb
baabaaa
bab
bb
bbabbaab

样例输出

1
0
3
0
0

答案

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#include <iostream>
using namespace std;

struct tNode
{
int cnt;
tNode *next[26];
void init()
{

cnt = 0;
for (int i = 0; i < 26; i++)
{
next[i] = NULL;
}
}
};

tNode *root = (tNode *)malloc(sizeof(tNode));

void insert(char *s)
{

int i = 0;
tNode * T = root;
while (s[i])
{
tNode *node = (tNode *)malloc(sizeof(tNode));
int x = s[i] - 'a';
if (T->next[x] == NULL)
{
node->init();
T->next[x] = node;
}
T = T->next[x];
T->cnt++;
i++;
}
}

void query(char *s)
{

int i = 0;
tNode * T = root;
while (s[i])
{
tNode *node = (tNode *)malloc(sizeof(tNode));
node->init();
int x = s[i] - 'a';
if (T->next[x] == NULL)
{
puts("0");
return;
}
node = T->next[x];
T = node;
i++;
}
cout << T->cnt << endl;
}

int main()
{

int n, m;
char str[20];
root->init();
cin >> n;
while (n--)
{
cin >> str;
insert(str);
}
cin >> m;
while (m--)
{
cin >> str;
query(str);
}
return 0;
}

收获与体会

  1. C++的Pointer使用不熟;
  2. 多叉树与二叉树操作类似,遇到此类问题不要恐惧;
  3. Trie树经典算法,以后可以深入研究;
  4. 建树过程需要掌握。