今天学字典树写的版本:
#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;
}