这是一道字典树的问题,直接利用模板,不过还要一个释放内存的函数!
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;
typedef struct trie
{
int v;
trie *next[10];
};
trie *root;
void creattrie(char *str)
{
int len,i,j;
trie *p=root,*q;
len=strlen(str);
for(i=0; i<len; i++)
{
int id=str[i]-'0';
if(p->next[id]==NULL)
{
q=(trie *)malloc(sizeof(trie));
q->v=1;
for(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;
}
int findtrie(char *str)
{
int len,i;
trie *p=root;
len=strlen(str);
for(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 deal(trie *T)
{
int i;
if(T==NULL)return 0;
for(i=0; i<10; i++)
{
if(T->next[i]!=NULL)
{
deal(T->next[i]);
}
}
free(T);
return 0;
}
int main()
{
int t,num;
int i;
int flag;
char str[20];
scanf("%d",&t);
while(t--)
{
flag=0;
root=(trie *)malloc(sizeof(trie));
for(i=0; i<10; i++)
root->next[i]=NULL;
scanf("%d",&num);
for(i=0; i<num; i++)
{
scanf("%s",str);
if(findtrie(str))
flag=1;
if(flag)
continue;
creattrie(str);
}
if(flag)
{
printf("NO\n");
}
else
{
printf("YES\n");
}
deal(root);
}
return 0;
}