poj 3630(Phone List)

题目链接:http://poj.org/problem?id=3630

    单纯的trie树。

code:

#include<cstring>
#include<cstdio>

char str[100] ;//模式串#define MAX 26 //字符集大小
typedef struct TrieNode{
    int count ; //记录该字符出现次数
    bool f ;
    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->count = 1 ;
    p->f = false ;
    for(i=0; i<MAX; i++){
        p->next[i] = NULL ;
    }
    return p ;
}

/*插入*/
bool InsertTrie(TrieNode **pRoot, char *s){
    int i, k, c ;
    TrieNode *p ;
    if(!(p=*pRoot)){
        p = *pRoot = CreateTrieNode() ;
    }
    i = c = 0 ;
    while(s[i]){
        k = s[i++] - '0' ; //确定branch
        if(p->next[k]){
            p->next[k]->count ++ ;
            if(p->next[k]->f)    return false ;
            c ++ ;
        }
        else
            p->next[k] = CreateTrieNode() ;
        p = p->next[k] ;
    }
    p->f = true ;
    if(c==strlen(s))    return false ;
    return true ;
}

//查找
int SearchTrie(TrieNode **pRoot, char *s){
    TrieNode *p ;
    int i , k ;

    if(!(p=*pRoot))
        return 0 ;
    i = 0 ;
    while(s[i]){
        k = s[i++] - '0' ;
        if(p->next[k]==NULL)    return 0 ;
        p = p->next[k] ;
    }
    return p->count ;
}
int main(){
    int n , i, t ;
    bool f ;
    scanf("%d", &t) ;
    while(t--){
        allocp = 0 ;
        TrieNode *root = NULL ;
        InitTrieRoot(&root) ;
        scanf("%d", &n) ;
        f = true ;
        for(i=0; i<n; i++){
            getchar() ;
            scanf("%s", str) ;
            if(f)
                f = InsertTrie(&root, str) ;
        }
        if(f)   printf("YES\n") ;
        else    printf("NO\n") ;
    }
    return 0 ;
}