Sicily 1426 Phone List

给出n个字符串，判断是否存在一个字符串是另一个字符串的前缀，典型的字典树的应用。

#include <iostream>
using namespace std;
#include <stdio.h>
#define MAX 10 
typedef struct TrieNode {
    int endFlag;
    struct TrieNode *next[MAX];
}TrieNode;
TrieNode Memory[1000010];
int allocp =0;
int flag;
void InitTrieRoot(TrieNode **pRoot) {
    *pRoot = NULL;
}
TrieNode *CreateTrieNode() {
    int i;
    TrieNode *p;
    p =&Memory[allocp++];
    p->endFlag = 0;
    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++] -'0';
        if(p->next[k]) {
            if (p->next[k]->endFlag==1||s[i]=='\0') {
                flag=1;
                return;
            }
        }
        else
            p->next[k] = CreateTrieNode();
        p = p->next[k];
    }
    p->endFlag=1;
}
int main()
{
    int t,n,i;
    char str[12];
    TrieNode *Root;

    scanf("%d",&t);
    while(t--) {
        flag=allocp=0;
        InitTrieRoot(&Root);
        scanf("%d",&n);
        getchar();
        for (i=0;i<n;i++) {
            scanf("%s",&str);
            if (flag==0)
                InsertTrie(&Root,str);
        }
        if (flag==0)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}