poj 2001 Shortest Prefixes (trie)

拿到这题，大致的画了画图，发现只要标记好每个节点出现的次数就好了。 

插入时，若字符存在，n++，若不存在建立新节点，他本身也为自己的前缀。

查找时，若next[k]->nCount!=1，说明其子节点有多个，无法确定唯一，则输出该字符，若nCount=1，则可以结束查找了。

code:

#include<cstdio>
#include<cstring>

char str[1001][21] ;
#define MAX 30 //字符集大小
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=0; 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' ; //确定branch
        if(p->next[k])
            p->next[k]->nCount ++ ;
        else
            p->next[k] = CreateTrieNode() ;
        p = p->next[k] ;
    }
}

//查找
void SearchTrie(TrieNode **pRoot, char *s){
    TrieNode *p ;
    p = *pRoot ;
    int i , k ;
    i = 0 ;
    while(s[i]){
        k = s[i++] - 'a' ;
        if(p->next[k]==NULL||p->next[k]->nCount==1){
            printf("%c", k+'a') ;
            break ;
        }
        else{
            p = p->next[k] ;
            printf("%c", k+'a') ;
        }
    }
    printf("\n") ;
}
int main(){
    int n = 0, i ;
    allocp = 0 ;

    TrieNode *root = NULL ;
    InitTrieRoot(&root) ;

    while(gets(str[n])&&str[n][0])
        InsertTrie(&root, str[n++]) ;
    for(i=0; i<n; i++){
        printf("%s ", str[i]) ;
        SearchTrie(&root, str[i]) ;
    }
    return 0 ;

}