POJ 1816 Wild Words

#include<cstdio>
#include<vector>
#include<cstring>
#define MAXN 100010
#define MAXM 110
using namespace std;
int ans[MAXN], n, m;
char s[25];
struct node
{
node *next[30];
vector<int> val;
node(){memset(next,0,sizeof(next));}
};
struct TRIE
{
int len;
node *root;
void pre(){root = new node();}
int idx(char ch)
{
if(ch=='?')return 'z'-'a'+1;
if(ch=='*')return 'z'-'a'+2;
else return ch-'a';
}
void insert(char *s, int id)
{
len=strlen(s);
node *p=root;
for(int i = 0; i < len; i++)
{
int c = idx(s[i]);
if(!p->next[c])
p->next[c]=new node();
p=p->next[c];
}
p->val.push_back(id);
}
void dfs(char *s, node *p, int pos)
{
if(s[pos]==0)
{
for(int i = 0; i < p->val.size(); i++)
{
ans[p->val[i]]=1;
}
if(p->next[idx('*')])
dfs(s,p->next[idx('*')],pos);
}
else
{
int c=idx(s[pos]);
if(p->next[c])dfs(s,p->next[c],pos+1);
if(p->next[idx('?')])dfs(s,p->next[idx('?')],pos+1);
if(p->next[idx('*')])
for(int i = pos; i <= len; i++)
dfs(s,p->next[idx('*')],i);
}
}
void find(char *s)
{
len=strlen(s);
memset(ans,0,sizeof(ans));
dfs(s,root,0);
bool mat=0;
for(int i = 1; i <= n; i++)
{
if(ans[i])
{
mat=1;
printf("%d ",i-1);
}
}
if(!mat)
printf("Not match");
printf("\n");
}
}trie;
int main()
{
scanf("%d%d",&n,&m);
trie.pre();
for(int i = 1; i <= n; i++)
{
scanf("%s",s);
trie.insert(s,i);
}
for(int i = 1; i <= m; i++)
{
scanf("%s",s);
trie.find(s);
}
return 0;
}