Wild Words POJ - 1816（模糊匹配问题，trie树+dfs ，有坑）

题目链接

题目大意：先给定n个文本串，再给定m个模式串，对每一个模式串，问其可以被哪些文本串匹配到，输出文本串。特殊规则为：文本串中’？‘字符可表示任意非空字符，’*'可表示空或任意一段字符串。

思路：首先是把文本串插入到字典树中， 并对结尾节点的序号做好标记， 但是题目的坑在于n个文本串中可能有一样的字符串， 这也是要分别计数的, (这里我wa了一上午orz 太菜了) , 我用pos[i]表示第i个字符串的结尾字符在插入字典树后的节点序号。
这些处理完之后，每输入一个模式串，我们用dfs来和字典树匹配，匹配成功的标志是：模式串走完 + 此时dfs到的字典树根节点是一个字符串的结尾节点。鳍鱼细节见代码注释~

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define INF 1000000
//题意：给定一些文本串，再给定模式串，问每个模式串可以被那些文本串匹配到。文本串中的？字符可表示任意字符，*可表示空或任意一段字符串
//文本串建字典树，由于文本串中可能有相同的串，记录好每个串的结尾节点序号pos[i]
//输入一个模式串后开始dfs，root从0开始，如果当前匹配成功或者是？,直接匹配模式串下一位.若是*，循环递归，见代码
const int maxn=2e6+10;
int n,T;
char s[25];
int tree[maxn][30];
int tot;
bool vis[maxn];//这个结尾的树串被匹配到了
bool isstr[maxn];
int pos[maxn];//第i个字符串的结尾是第pos[i]号节点，因为可能存在相同的树串，所以需要pos数组保证每个匹配到的都被记录
int Insert(char *s)
{
    int len=strlen(s);
    int root=0;
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'a';
        if(s[i]=='?') id=26;
        else if(s[i]=='*') id=27;
        if(!tree[root][id]) tree[root][id]=++tot;
        root=tree[root][id];
    }
    isstr[root]=true;
    return root;
}
void dfs(int root,int pos,int len)
{
    if(isstr[root] && pos==len)//匹配成功的标志，标记root节点已访问
    {
        vis[root]=1;//说明以这个root节点结尾的所有文本串都可以与当前的模式串匹配成功
    }
    int id=s[pos]-'a';
    if(id>=0 && tree[root][id]) dfs(tree[root][id],pos+1,len);
    if(tree[root][26]) dfs(tree[root][26],pos+1,len);
    if(tree[root][27])
    {
        for(int j=pos;j<=len;j++)//*可以为空，不匹配，所以j从pos开始
            dfs(tree[root][27],j,len);
    }
}
int main()
{
//    ios::sync_with_stdio(false);
//    cin.tie(0);
//    cout.tie(0);
    scanf("%d%d",&n,&T);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        pos[i]=Insert(s);
    }
    while(T--)
    {
        memset(vis,false,sizeof(vis));
        scanf("%s",s);
        //一个字符串在字典树中（带？和*）可以被哪些串匹配，串短，用dfs
        dfs(0,0,strlen(s));
        int flag=0;
        for(int i=1;i<=n;i++)
        {
            if(vis[pos[i]])//可能存在多个一样的字符串
            {
                flag=1;
                printf("%d ",i-1);
            }
        }
        if(!flag) printf("Not match");
        printf("\n");
    }
    system("pause");
    return 0;
}