题目大意:先给定n个文本串,再给定m个模式串,对每一个模式串,问其可以被哪些文本串匹配到,输出文本串。特殊规则为:文本串中’?‘字符可表示任意非空字符,’*'可表示空或任意一段字符串。
思路:首先是把文本串插入到字典树中, 并对结尾节点的序号做好标记, 但是题目的坑在于n个文本串中可能有一样的字符串, 这也是要分别计数的, (这里我wa了一上午orz 太菜了) , 我用pos[i]表示第i个字符串的结尾字符在插入字典树后的节点序号。
这些处理完之后,每输入一个模式串,我们用dfs来和字典树匹配,匹配成功的标志是:模式串走完 + 此时dfs到的字典树根节点是一个字符串的结尾节点。鳍鱼细节见代码注释~
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<unordered_map>
#include<math.h>
#include<set>
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;
}