zoj 1109 Language of FatMouse（字典树Trie）

这个应该算是字典树的经典应用了吧。

 

讲需要查找的内容存进字典树里~然后查找这个单词，就能把所存的data给调出来~

 

详情见大牛BLOG http://www.cnblogs.com/DreamUp/archive/2010/07/23/1783410.html

 

我的函数都是自己写的哦~~~嘻嘻~~~

 

WA了数次。因为node数组开小了，如果用malloc就不会出现这种情况了。。。可我不喜欢malloc。。。

 

我的node数组是从里面拿空间用的。

 

#include <stdio.h> #include <stdlib.h> #include <iostream> #include <string.h> #define MAX 100006 using namespace std; typedef struct Trie{ Trie *child[26]; int data; }Trie; Trie node[120000]; // I hate it!!! The size of the node should biger than 100006 = =. WA N times. int cou; void init() { cou = 1; memset(node,'/0',sizeof(node)); } void Build( char *a, int num) //Build the trie, magic ~ { Trie *head = &node[0]; int len = strlen(a); for(int i=0; i<len; i++) { if( head->child[a[i]-'a'] == NULL ) head->child[a[i]-'a'] = &node[cou++]; head = head->child[a[i]-'a']; } head->data = num; } int Find( char *a ) // Find the data in the trie. { Trie *head = &node[0]; int len = strlen(a); for(int i=0; i<len; i++) { head = head->child[a[i]-'a']; if( head == NULL ) return 0; } return head->data; } char eng[MAX][50],mice[MAX][50]; char tofind[50]; char str[50]; int main(void) { int count = 1; init(); while( gets(str) && strlen(str) ) { sscanf(str,"%s%s",eng[count],mice[count]); // = = . Fire station 's input format... count++; } for( int i=1; i<count; i++ ) Build( mice[i], i ); while( scanf("%s",tofind) != EOF ) { int ind = Find( tofind ); if( ind == 0 ) printf("eh/n"); else printf("%s/n",eng[ind]); } return 0; }