HDU 1671 Phone List(字典树or String)

今天学字典树写的版本：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef struct Trie
{
    int v;
    Trie *next[10];
}Trie;
Trie *root;
void CreateTrie(char *str)
{
    int len=strlen(str);
    Trie *p=root,*q;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'0';
        if(p->next[id]==NULL)
        {
            q=(Trie *)malloc(sizeof(Trie));
            q->v=1;
            for(int j=0;j<10;j++)
                q->next[j]=NULL;
            p->next[id]=q;
            p=p->next[id];
        }
        else
        {
            p->next[id]->v++;
            p=p->next[id];
        }
    }
    p->v=-1;
}
void deal(Trie* T)
{
    if(T==NULL) return ;
    for(int i=0;i<10;i++)
        if(T->next[i]!=NULL) deal(T->next[i]);
    free(T);
    return ;
}
int findTrie(char *str)
{
    Trie *p=root;
    int len=strlen(str);
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'0';
        p=p->next[id];
        if(p==NULL) return 0;///不是某个单词的前缀
        if(p->v==-1) return -1;///某个单词是其前缀
    }
    return -1;///是某个单词的前缀
}
int main()
{
    char word[15];
    int T,m;
    scanf("%d",&T);
    while(T--)
    {
        root=(Trie *)malloc(sizeof(Trie));
        for(int i=0;i<10;i++)
            root->next[i]=NULL;
        scanf("%d",&m);
        getchar();
        bool flag=true;
        while(m--)
        {
            scanf("%s",word);
            if(findTrie(word)==-1)
                flag=false;
            if(!flag) continue;///防止MLE
            CreateTrie(word);
        }
        if(flag) printf("YES\n");
        else printf("NO\n");
        deal(root);
    }
    return 0;
}

以前学STL写的版本：

#include <cstdio>
#include <vector>
#include <string>
#include <algorithm>
#include <iostream>
using namespace std;
int main()
{
    vector<string>v;
    char ch[10000];
    string s;
    int n,t;
    cin>>t;
    while(t--)
    {
        cin>>n;
        v.clear();
        for(int i=0; i<n; i++)
        {
            scanf("%s",ch);
            s=ch;
            v.push_back(s);
        }
        sort(v.begin(),v.end());
        bool flag=true;
        for(int i=0; i<n-1; i++)
            if(v[i+1].find(v[i])==0)
            {
                printf("NO\n");
                flag=false;
                break;
            }
        if(flag) printf("YES\n");
    }
    return 0;
}