pku 1816 Wild Words

字典树+深搜。。

这个题和hdu上的T9有点相似，不过应该比那个要稍难一点，这个需要处理？和*的情况，还要处理模式串相同的情况。。

这是我们前几天比赛的题目，当时没有做出来，开始的时没有想起来用字典树，后来队友说用字典树就写了一下。

写着写着感觉还可以，把？和*分别看成是next的第26和27，建树的时候很简单。之后查询的时候要考虑全面，

对于*进行特殊的处理，它可能是一个字符也可能是多个字符也可能没有字符。。

还有一点就是对于相同的模式串， 要标记一下。

例如

2 1

abc

abc

abc

刚开始的时候把temp->num=-1;

Insert第0个模式串的时候把当前的temp->num=0;

之后插入第一个的时候把father[1]=0;然后把temp->num=1;这样就构成了一个线性的链了。

查询结束的时候 按线性查找所有的数 都加入到数组里面，对数组排序即可。

代码：

# include<stdio.h>
# include<string.h>
# include<stdlib.h>
# define MAX 28
struct Trie{
int num;
Trie *next[MAX];
};
char keyword[10];
char str[25];
int count[400005],k,father[100005],len;
int cmp(const void *a,const void *b)
{
return *(int *)a - *(int *)b;
}
Trie *NewTrie()
{
int i;
Trie *temp=new Trie;
temp->num=-1;
for(i=0;i<MAX;i++)
temp->next[i]=NULL;
return temp;
}
void Insert(Trie *p,char s[],int ans)
{
int i;
Trie *temp=p;
i=0;
while(s[i])
{
if(s[i]>='a' && s[i]<='z')
{
if(temp->next[s[i]-'a']==NULL) temp->next[s[i]-'a']=NewTrie();
temp=temp->next[s[i]-'a'];
}
else
{
if(s[i]=='?')
{
if(temp->next[26]==NULL) temp->next[26]=NewTrie();
temp=temp->next[26];
}
else
{
if(temp->next[27]==NULL) temp->next[27]=NewTrie();
temp=temp->next[27];
}
}
i++;
}
if(temp->num==-1) temp->num=ans;
else
{
father[ans]=temp->num;
temp->num=ans;
}
}
void dfs(Trie *p,int i)
{
int j;
Trie *temp=p;
if(str[i]==0)
{
if(temp->num!=-1)
{
k++;
count[k]=temp->num;
}
while(temp->next[27]!=NULL )
{
if(temp->next[27]->num!=-1)
{
k++;
count[k]=temp->next[27]->num;
}
temp=temp->next[27];
}
return;
}

if(temp->next[str[i]-'a']!=NULL) dfs(temp->next[str[i]-'a'],i+1);
if(temp->next[26]!=NULL) dfs(temp->next[26],i+1);
if(temp->next[27]!=NULL)
{
for(j=i;j<=len;j++)
dfs(temp->next[27],j);
}

}
int main()
{
int i,n,m,j,ans;
Trie *p;
while(scanf("%d%d",&n,&m)!=EOF)
{
p=NewTrie();
for(i=0;i<=n;i++)
father[i]=i;
for(i=0;i<n;i++)
{
scanf("%s",keyword);
Insert(p,keyword,i);
}
while(m--)
{
scanf("%s",str);
len=strlen(str);
k=0;
dfs(p,0);
if(k==0) printf("Not match\n");
else
{
ans=k;
for(i=1;i<=ans;i++)
{
j=count[i];
while(father[j]!=j)
{
k++;
count[k]=father[j];
j=father[j];
}
}
qsort(count+1,k,sizeof(count[1]),cmp);
count[0]=-1;
for(i=1;i<=k;i++)
{
if(count[i]!=count[i-1])
printf("%d ",count[i]);
}
printf("\n");
}
}
}
return 0;
}