HDU1251-字典树模版

最新推荐文章于 2021-11-09 00:20:03 发布

转载最新推荐文章于 2021-11-09 00:20:03 发布 · 444 阅读

#include<stdio.h>
#define MAX 26
typedef struct TrieNode
{
    int nCount;
    struct TrieNode *next[MAX];
}TrieNode;

TrieNode Memory[1000000];
int allocp=0;

void InitTrieRoot(TrieNode **pRoot)
{
    *pRoot=NULL;
}        

TrieNode *CreateTrieNode()
{
    int i;
    TrieNode *p;
    
    p=&Memory[allocp++];
    p->nCount=1;
    for(i=1;i<MAX;i++)
    {
        p->next[i]=NULL;
    }    
    return p;
}    

void InsertTrie(TrieNode **pRoot,char *s)
{
    int i,k;
    TrieNode *p;
    
    if(!(p=*pRoot))
      p=*pRoot=CreateTrieNode();
    i=0;
    while(s[i])
    {
        k=s[i++]-'a';
        if(p->next[k]) p->next[k]->nCount++;
        else p->next[k]=CreateTrieNode();
        p=p->next[k];
    }    
}    
int SearchTrie(TrieNode **pRoot,char *s)
{
    TrieNode *p;
    int i,k;
    if(!(p=*pRoot))
      return 0;
    i=0;
    while(s[i])
    {
        k=s[i++]-'a';
        if(p->next[k]==NULL) return 0;
        p=p->next[k];
    }    
    return p->nCount;
}    
int main()
{
    char str[11];
    TrieNode *Root=NULL;
    InitTrieRoot(&Root);
    while(gets(str)&&str[0])
    {
        InsertTrie(&Root,str);
    }    
    while(gets(str))
    {
        printf("%d\n",SearchTrie(&Root,str));
    }    
    return 0;
}