百度suggestion功能的Trie实现(附代码)

今年的百度实习生笔试题中有个题是：

相信大家都使用过百度搜索框的suggestion功能，百度搜索框中的suggestion提示功能如何实现？请给出实现思路和主要的数据结构、算法。有什么优化思路可以使得时间和空间效率最高？

这个题的基础实现方法是使用Trie树， 原理部分摘用别的. (代码部分我增加了findTips功能， 这里采用方法1,空间要求最大 26^n,,若是汉字，那不能使用了。)

树的结构可以有这三种实现方法 ：

一般有三种方法：

1、对每个结点开一个字母集大小的数组，对应的下标是儿子所表示的字母，内容则是这个儿子对应在大数组上的位置，即标号；

2、对每个结点挂一个链表，按一定顺序记录每个儿子是谁；

3、使用左儿子右兄弟表示法记录这棵树。

三种方法，各有特点。第一种易实现，但实际的空间要求较大；第二种，较易实现，空间要求相对较小，但比较费时；第三种，空间要求最小，但相对费时且不易写。
http://zhangzhibiao02005.blog.163.com/blog/static/373678202010667524646/
昨天在北邮人bbs上逛时，看到有人在讨论字符窜提示的问题，有人回帖用trie树来做，而我以前也听过trie树，不过想系统学习下，于是大概用了一个下午的时间来学习trie树。网上有很多关于trie树的资料，其核心思想就时拿空间换取时间，很适合搜索前缀子窜，时间效率比较高，主要是占用空间比较大。为什么会这样呢？看下trie树的结构就知道拉。

如图所示，该trie树存有abc、d、da、dda四个字符串，如果是字符串会在节点的尾部进行标记。没有后续字符的branch分支指向NULL



已知n个由小写字母构成的平均长度为10的单词,判断其中是否存在某个串为另一个串的前缀子串。下面对比3种方法：

1. 最容易想到的：即从字符串集中从头往后搜，看每个字符串是否为字符串集中某个字符串的前缀，复杂度为O(n^2)。

2. 使用hash：我们用hash存下所有字符串的所有的前缀子串。建立存有子串hash的复杂度为O(n*len)。查询的复杂度为O(n)*
O(1)= O(n)。

3. 使用trie：因为当查询如字符串abc是否为某个字符串的前缀时，显然以b,c,d....等不是以a开头的字符串就不用查找了。所以建立trie的复杂度为O(n*len)，而建立+查询在trie中是可以同时执行的，建立的过程也就可以成为查询的过程，hash就不能实现这个功能。所以总的复杂度为O(n*len)，实际查询的复杂度只是O(len)

解释一下hash为什么不能将建立与查询同时执行，例如有串：911，911456输入，如果要同时执行建立与查询，过程就是查询911，没有，然后存入9、91、911，查询911456，没有然后存入9114、91145、911456，而程序没有记忆功能，并不知道911在输入数据中出现过。所以用hash必须先存入所有子串，然后for循环查询。

而trie树便可以，存入911后，已经记录911为出现的字符串，在存入911456的过程中就能发现而输出答案；倒过来亦可以，先存入911456，在存入911时，当指针指向最后一个1时，程序会发现这个1已经存在，说明911必定是某个字符串的前缀，

trie树的简单实现(增加了findTips功能）：

#include <iostream>
#include <cstdlib>
#include <string>
#include <string.h>
#include <vector>

using namespace std;

const int branchNum = 26;

struct TrieNode
{
bool isStr;
TrieNode *next[branchNum];
TrieNode()
:isStr(false)
{
memset(next , 0  , sizeof(next));
}
};

class Trie
{
public:
Trie();
void insert(const char *word);
bool search(const char *word);
void deleteTrie(TrieNode *root);
vector<string> findTips(const char *prefix); //here
void findTips(TrieNode *root , string track , vector<string> &tips); //here
private:
TrieNode *root;
};

Trie::Trie()
{
root = new TrieNode(); //绗竴涓猲ode鏃犵敤
}

void Trie::insert(const char *word)
{
TrieNode *location = root;
while(*word) {
if(location->next[*word - 'a'] == NULL) {
TrieNode *tmp = new TrieNode();
location->next[*word - 'a'] = tmp;
}
location = location->next[*word - 'a'];
word++;
}
location->isStr = true;
}

bool Trie::search(const char *word)
{
TrieNode *location = root;
while(*word && location) {
location = location->next[*word - 'a'];
word ++;
}
return (location != NULL && location->isStr);
}

void Trie::deleteTrie(TrieNode *root)
{
for(int i = 0 ; i < branchNum ; i++) {
if(root->next[i] != NULL) {
deleteTrie(root->next[i]);
}
}
delete root;
}

vector<string> Trie::findTips(const char *prefix)
{
vector<string> tips;
string track(prefix);
int i;
struct TrieNode *node;

for(i = 0 ,node = root ; i < strlen(prefix) && node != NULL ; i++) {
node = node->next[prefix[i] - 'a'];
}
if(node == NULL && i != strlen(prefix))
return tips; //empty

findTips(node, track , tips);
return tips;
}

void Trie::findTips(TrieNode *root,string track , vector<string> &tips) //track长度最多为最长字符串长度，
{

if(root == NULL)
return;

if(root->isStr)
tips.push_back(track);

for(int i = 0 ; i < branchNum ; i++) {
findTips(root->next[i] , track + (char)('a'+i) , tips);
}
}

int main()
{
Trie t;
t.insert("a");
t.insert("abandon");
t.insert("abandoned");
t.insert("abashed");
if(t.search("abashed"))
cout << "true" << endl;

vector<string> tips = t.findTips("ab");
cout << "ab" << " tips are :(" << tips.size() <<")" <<endl;
for(int i = 0 ; i < tips.size() ; i ++)
cout << tips[i] << endl;

}

运行后出来结果：

true

ab tips are :(3)

abandon

abandoned

abashed

至于优化的话，也许在Trie树的实现就可以进行优化。向这种实现方式的化，每个节点就得包括所有的字符集。如果换成汉字，那就是每个结点上千个了。

注意：对于汉字的话，可以弄一个索引，比如0代表啊，1代码阿，等等 ， 就无法使用了。

树的深度就是最长的单词长度 。。

比如：http://linux.thai.net/~thep/datrie/datrie.html#Double双数组的实现方式