题目传送门:
http://acm.hdu.edu.cn/showproblem.php?pid=1251
Trie树是一种特殊的多叉树结构,又称字典树或前缀树。特点是占用内存大(可以尝试左儿子右兄弟存储),但是可以很好的节约时间,时间复杂度为O(N),N为查找或插入的字符串长度。
题目中统计含某一前缀的单词数正是Trie树的典型应用。
Trie树的定义:
typedef struct Trie{
//记录含有该前缀的单词数目
int cnt;
/*指针域,表示26个小写字母,NULL意味着字典中并没有以该前缀开头的单词*/
struct Trie *next[26];
} Trie,*pTrie;
基本操作:
创建Trie树:
//返回根节点指针
pTrie Init(void)
{
pTrie p = (pTrie)malloc(sizeof(Trie));
p -> cnt = 0;
for (int i = 0; i < 26; ++i)
{
p -> next[i] = NULL;
}
return p;
}
添加单词:
void Add(pTrie p, char *s)
{
int len = strlen(s);
for (int i = 0; i < len; ++i)
{
int id = s[i] - 'a';
// 从首字母开始扫描,当遇到新的前缀时新增结点
if (p->next[id] == NULL) {
pTrie q = (pTrie)malloc(sizeof(Trie));
q -> cnt = 1;
int j;
for (j = 0; j < 26; j ++) {
q -> next[j] = NULL;
}
p -> next[id] = q;
p = p -> next[id];
} else { //当前前缀已存在,cnt ++
p -> next[id] -> cnt ++;
p = p -> next[id];
}
}
return ;
}
查找:
//树的遍历,返回以s为前缀的单词数目
int Find(pTrie p, char *s)
{
int len = strlen(s);
for (int i = 0; i < len; ++i)
{
int id = s[i] - 'a';
if (p -> next[id] == NULL) {
return 0;
} else {
p = p->next[id];
}
}
return p -> cnt;
}
有了这些基本操作,题目也就简单了。
AC代码
// TRIE
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
using namespace std;
typedef struct Trie{
int cnt;
struct Trie *next[26];
} Trie,*pTrie;
pTrie Init(void)
{
pTrie p = (pTrie)malloc(sizeof(Trie));
p -> cnt = 0;
for (int i = 0; i < 26; ++i)
{
p -> next[i] = NULL;
}
return p;
}
void Add(pTrie p, char *s)
{
int len = strlen(s);
for (int i = 0; i < len; ++i)
{
int id = s[i] - 'a';
if (p->next[id] == NULL) {
pTrie q = (pTrie)malloc(sizeof(Trie));
q -> cnt = 1;
int j;
for (j = 0; j < 26; j ++) {
q -> next[j] = NULL;
}
p -> next[id] = q;
p = p -> next[id];
} else {
p -> next[id] -> cnt ++;
p = p -> next[id];
}
}
return ;
}
int Find(pTrie p, char *s)
{
int len = strlen(s);
for (int i = 0; i < len; ++i)
{
int id = s[i] - 'a';
if (p -> next[id] == NULL) {
return 0;
} else {
p = p->next[id];
}
}
return p -> cnt;
}
int main(int argc, char const *argv[])
{
pTrie p = Init();
char s[100];
while (gets_s(s),s[0] != '\0') {
Add(p,s);
}
while (~scanf("%s",s)) {
cout << Find(p,s) << endl;
}
return 0;
}