hdu1305 字典树

这题我开始想的简单了，WA一次，然后看disscuss里有人说输入时长度从小到大的，然后我信了。然后开始while(1) WA;然后我尝试先放如数组。后来对了；

discuss里面果然不能太相信。

根据出现的次数来判断是否为前缀。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
struct trie
{
trie *next[2];
int sum;
};
trie *root;
void creattrie()
{
root=(trie*)malloc(sizeof(trie));
for(int i=0;i<2;i++)
{
root->next[i]=NULL;
}
root->sum=0;
}
void insert(char *str)
{
int i,j,cnt=0;
int len=strlen(str);
trie *p=root,*q;
for(i=0;i<len;i++)
{
int id=str[i]-'0';
if(p->next[id]==NULL)
{
q=(trie*)malloc(sizeof(trie));
for(j=0;j<2;j++)
q->next[j]=NULL;
q->sum=0;
p->next[id]=q;
}
p=p->next[id];
p->sum++;
}
}

int query(char *str)
{
int i,j;
int cnt=0;
int len=strlen(str);
trie *p=root;
for(i=0;i<len;i++)
{
int id=str[i]-'0';
p=p->next[id];
}
if(p->sum>1)//判断到该字符串结束时，现在的sum是否超过2，如果是，那就代表后面有以这个为前缀的词
cnt=1;
if(cnt)
return 1;
return 0;
}

int main()
{
int i,j,flag,ff=0,ret,count;
char str[100][30];
while(gets(str[0])!=NULL)
{
if(str[0][0]=='9')break;
count=1;
creattrie();
insert(str[0]);
while(gets(str[count]))
{
if(str[count][0]=='9')break;
insert(str[count]);
count++;
}
flag=0;
for(i=0;i<count;i++)
{
flag=query(str[i]);
if(flag)break;
}
if(flag)printf("Set %d is not immediately decodable\n",++ff);
else printf("Set %d is immediately decodable\n",++ff);
}
return 0;
}