【POJ 1816 Wild Words】 字典树(Trie树)

POJ1816
本题的题意是给你n个模式串和m个匹配串，模式串中有$？和*$两种字符，？可以匹配任意一种字符，$*$可以匹配任意个字符，问每种匹配串可以和之前哪些模式串匹配。
我们先构建好字典树，对于每个匹配串实现find，find的时候用类似dfs的写法，如果当前节点的字符或者?存在，直接dfs下一个位置的字符，如果当前字符对应的位置有$*$存在，那么就从匹配串之后的每一个位置进行往下dfs，因为可以匹配任意个字符。而dfs计数的条件是匹配串匹配结束而且达到模式串某个结尾标记。则在这个模式串上做标记。需要注意的细节是此时不能直接return,因为有可能此时的字符串和剩下的匹配串还能匹配，类似模式串A ： AA** 模式串B： AA*** 匹配串为AAA ,那么匹配到A字符串之后则不能停止，要继续往下搜索。
POJ1816代码

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<sstream>
using namespace std;
const int maxn =2e6+5;
int tree[maxn][30];
bool flag[maxn];
bool vis[maxn];
int tot;
int insert_(char *str)
{
   int len=strlen(str);
   int root=0;
   for(int i=0;i<len;i++)
   {
       int id;
        if(str[i]=='*') id=26;//*号存在第26位
        else if(str[i]=='?') id=27;//？存在第27位
        else   id=str[i]-'a';
       if(!tree[root][id]) tree[root][id]=++tot;
       root=tree[root][id];
   }
   flag[root]=true;//结尾符标记
   return root;
}
void find_(char *str,int pos1,int pos2)//pos1 模式串下标 pos2 匹配串下标
{
    int len=strlen(str);
    if(pos2==len&&flag[pos1])//达到条件则对该字符串进行标记
    {
        vis[pos1]=1;//对该单词的下标进行标记
    }
    int root=pos1;
    int id=str[pos2]-'a';
    if(tree[root][id])//存在相同字符
    {
        int tmp=tree[root][id];
        find_(str,tmp,pos2+1);
    }
    if(tree[root][27])//存在？
    {
        int tmp=tree[root][27];
        find_(str,tmp,pos2+1);
    }
    if(tree[root][26])//存在*
    {
        int tmp=tree[root][26];
        for(int j=pos2;j<=len;j++)//对之后的每一个位置进行dfs,要注意从pos2开始，因为*号可以不匹配
            find_(str,tmp,j);
    }
    return ;
}
char ss[maxn];
int pos[maxn];
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)
    {
        scanf("%s",ss);
        pos[i]=insert_(ss);//插入时顺带获取下标
    }
    while(m--)
    {
        int flag=0;
        scanf("%s",ss);
        find_(ss,0,0);
        for(int i=0;i<n;i++)
        {
            if(vis[pos[i]])
            {
                flag=1;
                 printf("%d ",i);
            }
        }
        for(int i=0;i<n;i++)
        {
            if(vis[pos[i]])
            {
                vis[pos[i]]=0;
            }
        }
        if(flag==0) printf("Not match");
        printf("\n");
    }
    return 0;
}