程序员面试金典——解题总结： 9.18高难度题 18.8---拓展： 实现一个Trie树

#include <iostream>#include <stdio.h>#include <string>#include <vector>using namespace std;/*问题:请实现一个trie树，并执行查找。首先实现trie树：trie树，本质上是一颗多叉树，每个结点是一个字符，如果在某个结点处对应的标记说从根节点到当前结点组成的内容是一个字符串，                就表明找到了一个字符串。本质是将一个字符串的各个字符作为结点插入到树中。输入:10(字符串数组中元素个数n) cao(待查找字符串)cai cao li lan cha chang wen chao yun yang10(字符串数组中元素个数n) cacai cao li lan cha chang wen chao yun yang输出:yes(能够查找到,不能查找到输出no)no*///前缀树结点包含：孩子结点指针数组，从根节点到当前结点是否是一个字符串的标记【结点的字符，体现在孩子结点指针数组的下标上,下标index = 字符 - 'a'】const int BRANCH_NUM = 26;//26个英文字符class TrieNode{public:TrieNode(){_isString = false;memset(_childs , NULL , sizeof(_childs) );}bool _isString;TrieNode* _childs[BRANCH_NUM];};//前缀树包含了：根节点，查找，删除，销毁等成员和函数class TrieTree{public:TrieTree(){}~TrieTree(){deleteTrie(_root);}//插入一个字符串:遍历每个字符，插入到对应子树上，当所有字符处理玩，这只最后一个字符对应结点的字符串标记为真void insert(char* str){if(NULL == str || NULL == _root){return;}TrieNode* curNode = _root;while( (*str) != '\0'){//找到对应子树int index = (*str) - 'a';if(index < 0 || index >= BRANCH_NUM){continue;}//寻找对应子树，如果为空，就新建子树对应结点TrieNode* childNode = curNode->_childs[index];if(NULL == childNode){TrieNode* node = new TrieNode();curNode->_childs[index] = node;}str++;//设置当前结点继续往下走curNode = curNode->_childs[index];}//处理完成后，设置最后结点对应的字符串标记为真curNode->_isString = true;}//查找某个字符串:对每个字符继续向下遍历对应的结点，直到字符串为空时，此时如果最后一个字符对应的结点的字符串标记如果为真，就说明查找到bool search(char* str){if(NULL == str || NULL == _root ){return false;}TrieNode* curNode = _root;while( (*str) != '\0'){int index = (*str) - 'a';if(index < 0 || index >= BRANCH_NUM){continue;}//寻找对应子树，如果为空，就说明没有找到TrieNode* childNode = curNode->_childs[index];if(NULL == childNode){return false;}str++;curNode = childNode;}return curNode->_isString;}//删除树,递归删除：先递归，最后删除结点void deleteTrie(TrieNode* root){if(NULL == root){return;}for(int i = 0 ; i < BRANCH_NUM ; i++ ){if( root->_childs[i] != NULL ){deleteTrie(root->_childs[i]);}}delete root;root = NULL;}TrieNode* _root;};void process(){int strNum;string searchStr;string str;while(cin >> strNum >> searchStr){//scanf("%s" , searchStr);TrieTree trieTree;TrieNode* root = new TrieNode();trieTree._root = root;for(int i = 0 ; i < strNum ; i++){//memset(str , NULL , sizeof(str));//scanf("%s" , str);cin >> str;//每输入一个字符串，就插入到前缀树中trieTree.insert((char*)str.data());}//接下来执行咋找bool isFind = trieTree.search((char*)searchStr.c_str());if(isFind){cout << "yes" << endl;}else{cout << "no" << endl;}}}int main(int argc , char* argv[]){process();getchar();return 0;}