题意:给你n个字符串,问其中是否存在字符串是其他某个字符串的前缀。
这是一道字典树模板题。由于之前只会用静态分配内存的方法来做,也就是用二维数组代替指针来建树,这种方法容易爆内存。后来学了指针建树,这种方法需要释放内存。
题目链接:点击打开链接
学洗了这位大牛的博:点击打开链接
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<stdlib.h>
using namespace std;
const int maxm=10;
typedef struct trie
{
int v;
trie *next[maxm];
}trie;
trie *root;
char str[22];int len;
void creat();
void deal(trie *t);
int find();
int main()
{
int n,i,j,k,sum,t,flag;
scanf("%d",&t);
while(t--)
{
flag=0;
root=(trie *)malloc(sizeof(trie));
for(i=0;i<10;i++)
root->next[i]=NULL;
scanf("%d",&n);
for(i=0;i<n;i++)
{
scanf("%s",str);
len=strlen(str);
if(find()==-1)
flag=1;
if(flag)
continue;
creat();
}
if(flag)
printf("NO\n");
else
printf("YES\n");
deal(root);
}
return 0;
}
void creat()
{
int i,j,k,id;
trie *p=root,*q;
for(i=0;i<len;i++)
{
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 find()
{
int i,j,k,id;
trie *p=root;
for(i=0;i<len;i++)
{
id=str[i]-'0';
p=p->next[id];
if(p==NULL)
return 0;
if(p->v==-1)
return -1;
}
return -1;
}
void deal(trie *t)
{
int i;
if(t==NULL)
return;
for(i=0;i<10;i++)
{
if(t->next[i]!=NULL)
deal(t->next[i]);
}
free(t);
return;
}