hdu 1251

字典树!!! 题目代码后后是模板!!! #include #include #include #define MAX 26 using namespace std; struct trie { trie* next[MAX]; int v; }; int main() { char s1[15]; trie *root=(trie*)new(trie); for(int i=0;inext[i]=NULL; while(1) { trie *p=root,*q; gets(s1); int len=strlen(s1); if(strcmp(s1,"")==0) { break; } for(int i=0;inext[id]==NULL) { q=(trie *)new(trie); q->v=1; for(int j=0;jnext[j]=NULL; p->next[id]=q; p=p->next[id]; } else { p->next[id]->v++; p=p->next[id]; // printf("%d/n",p->v); } } //p->v=-1; } while(scanf("%s",s1)!=EOF) { trie *p=root,*q; int len=strlen(s1); p=root; int flag=0; for(int i=0;inext[id]; if(p==NULL) { flag=1; printf("0/n"); break; } /* else if(p->v==-1) { flag=1; printf("1/n"); break; }*/ } if(flag==0) printf("%d/n",p->v); } return 0; } Trie的数据结构定义: #define MAX 26 typedef struct Trie { Trie *next[MAX]; int v; //根据需要变化 }; Trie *root; next是表示每层有多少种类的数,如果只是小写字母,则26即可,若改为大小写字母,则是52,若再加上数字,则是62了,这里根据题意来确定。 v可以表示一个字典树到此有多少相同前缀的数目,这里根据需要应当学会自由变化。 Trie的查找(最主要的操作): (1) 每次从根结点开始一次搜索; (2) 取得要查找关键词的第一个字母,并根据该字母选择对应的子树并转到该子树继续进行检索;   (3) 在相应的子树上,取得要查找关键词的第二个字母,并进一步选择对应的子树进行检索。    (4) 迭代过程……    (5) 在某个结点处,关键词的所有字母已被取出,则读取附在该结点上的信息,即完成查找。 这里给出生成字典树和查找的模版: 生成字典树: void createTrie(char *str) { int len = strlen(str); Trie *p = root, *q; for(int i=0; inext[id] == NULL) { q = (Trie *)malloc(sizeof(Trie)); q->v = 1; //初始v==1 for(int j=0; jnext[j] = NULL; p->next[id] = q; p = p->next[id]; } else { p->next[id]->v++; p = p->next[id]; } } p->v = -1; //若为结尾,则将v改成-1表示 } 接下来是查找的过程了: int findTrie(char *str) { int len = strlen(str); Trie *p = root; for(int i=0; inext[id]; if(p == NULL) //若为空集,表示不存以此为前缀的串 return 0; if(p->v == -1) //字符集中已有串是此串的前缀 return -1; } return -1; //此串是字符集中某串的前缀 } 对于上述动态字典树,有时会超内存,比如 HDOJ 1671 Phone List,这是就要记得释放空间了: int dealTrie(Trie* T) { int i; if(T==NULL) return 0; for(i=0;inext[i]!=NULL) deal(T->next[i]); } free(T); return 0; }

你可能感兴趣的:(acm---字符串)