【字典树应用】联想用户最想输入的词汇

1 struct TrieNode
2 {
3     struct TrieNode* next[26];
4     unsigned int     count;
5 };

1 TrieNode* CreateTrieNode()
2 {
3     TrieNode* head = (TrieNode*)malloc(sizeof(TrieNode));
4     memset(head, 0, sizeof(TrieNode));
5     return head;
6 }

 1 //遍历字符串，如果字符串未全部写入路径则查找失败，否则返回词频
 2 unsigned int FindTrieNode(TrieNode* head, char* str)
 3 {
 4     if(NULL == head)
 5         return 0;
 6     TrieNode* tmphead = head;
 7     int i = 0, cnt = 0;
 8     while(str[i])
 9     {
10         cnt = str[i] - ‘a‘;
11         if(NULL != tmphead->next[cnt])
12         {
13             tmphead = tmphead->next[cnt];
14             i++;
15         }
16         else
17         {
18             return 0;
19         }
20     }
21     return tmphead->count;
22 }

 1 //遍历整个字符串，创建不存在的字典树路径，并将整个路径的词频++
 2 //如果malloc失败，表示进程内存不足，插入字符串失败
 3 BOOL InsertTrieNode(TrieNode* head, char* str)
 4 {
 5     if(NULL == head)
 6     {
 7         head = (TrieNode*)malloc(sizeof(TrieNode));
 8         memset(head, 0, sizeof(TrieNode));
 9     }
10     int i = 0, cnt = 0;
11     TrieNode* tmphead = head;
12     while(str[i])
13     {
14         cnt = str[i] - ‘a‘;
15         if(NULL == tmphead->next[cnt])
16         {
17             if(NULL != (tmphead->next[cnt] = (TrieNode*)malloc(sizeof(TrieNode))))
18                 memset(tmphead->next[cnt], 0, sizeof(TrieNode));
19             else
20                 return FALSE;
21         }
22         tmphead = tmphead->next[cnt];
23         i++;
24         tmphead->count++;
25     }
26     return TRUE;
27 }

 1 //更新用户输入频率
 2 void UpdateFrequence(TrieNode* head, char* str)
 3 {
 4     int i = 0, cnt = 0;
 5     while(str[i])
 6     {
 7         cnt = str[i] - ‘a‘;
 8         head->next[cnt]->count++;
 9         head = head->next[cnt];
10         i++;
11     }
12 }

 1 //根据已输入的字符串联想出词频最高的一个字符串，即最有可能是用户想要输入的完整字符串。
 2 //已输入的字符串必须已经插入字典树
 3 //如果存在路径包含的情况，则总是返回最长路径：比如先插入pretty，再插入prettygirl，输入pr则联想词汇为prettygirl
 4 char* GetWantedWord(TrieNode* head, char* szSrc, char* szDes, size_t stDesLen)
 5 {
 6     if(NULL == head || NULL == szSrc || stDesLen <= strlen(szSrc))
 7         return szSrc;
 8     TrieNode* tmphead = head;
 9     int i = 0, cnt = 0;
10     unsigned int uiMax = 0, len = 0;
11     char         cMax = ‘a‘;
12     while(szSrc[i])
13     {
14         cnt = szSrc[i] - ‘a‘;
15         tmphead = tmphead->next[cnt];
16         szDes[i] = szSrc[i];
17         len++;
18         i++;
19     }
20     while(1)
21     {
22         uiMax = 0;
23         cMax = ‘a‘;
24         for(int j = 0; j < 26; j++)
25         {
26             if(NULL != tmphead->next[j] && tmphead->next[j]->count > uiMax)
27             {
28                 uiMax = tmphead->next[j]->count;
29                 cMax = j + ‘a‘;
30             }
31         }
32         if(len < stDesLen && uiMax > 0)
33             szDes[i] = cMax;
34         else
35         {
36             szDes[i] = ‘\0‘;
37             return szDes;
38         }
39         cnt = cMax - ‘a‘;
40         tmphead = tmphead->next[cnt];
41         len++;
42         i++;
43     }
44     szDes[i] = ‘\0‘;
45     return szDes;
46 }