唉,难道说在大牛的肩膀上做题会容易么-,-
2826252 | 2012-03-28 00:36:32 | Accepted | 2876 | C++ | 250 | 2688 | zisu_123 |
之前那题不需要加上dealTrie释放空间,这题不行,不然MLE的。
不过加上总是一个好习惯呗。
模版-,-Orz
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define MAX 10
typedef struct Trie{
Trie *next[MAX];
int n;
}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->n=1;
for(int j=0;j<MAX;j++)
q->next[j]=NULL;
p->next[id]=q;
p=p->next[id];
}
else
{
p->next[id]->n++;
p=p->next[id];
}
}
p->n=-1;
}
int findTrie(char *str)
{
int len=strlen(str);
Trie *p=root;
for(int i=0;i<len;i++)
{
int id=str[i]-'0';
p=p->next[id];
if(p==NULL) return 0;
if(p->n==-1)
return -1;
}
return -1;
}
int dealTrie(Trie* T)
{
int i;
if(T==NULL)
return 0;
for(i=0;i<MAX;i++)
if(T->next[i]!=NULL)
dealTrie(T->next[i]);
free(T);
return 0;
}
int main()
{
int n,m;
char num[11];
scanf("%d",&n);
while(n--)
{
int flag=0;
root=(Trie *)malloc(sizeof(Trie));
for(int i=0;i<MAX;i++)
root->next[i]=NULL;
scanf("%d",&m);
for(int i=0;i<m;i++)
{
scanf("%s",num);
if(findTrie(num)==-1) flag=1;
createTrie(num);
}
if(flag) printf("NO\n");
else printf("YES\n");
dealTrie(root);
}
return 0;
}