Wild Words

http://poj.org/problem?id=1816 

#include <iostream>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <map>
//#include<bits/stdc++.h>
using namespace std;

#define sfi(i) scanf("%d",&i)
#define sfl(i) scanf("%lld",&i)
#define sfs(i) scanf("%s",(i))
#define pri(i) printf("%d\n",i)
#define sff(i) scanf("%lf",&i)
#define ll long long
#define ull unsigned long long
#define mem(x,y) memset(x,y,sizeof(x))
#define INF 0x3f3f3f3f
#define eps 1e-10
#define PI acos(-1.0)
#define lowbit(x) ((x)&(-x))
#define zero(x) (((x)>0?(x):-(x))<eps)
#define fl() printf("flag\n")
#define MOD(x) ((x%mod)+mod)%mod
#define endl '\n'
#define pb push_back
#define FAST_IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)

/*
//---------------------------------------------------------
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
#include <ext/pb_ds/priority_queue.hpp>
using namespace __gnu_pbds;
*/

//gp_hash_table<string,int>mp2;
//__gnu_pbds::priority_queue<int>q;//因为放置和std重复，故需要带上命名空间
//__gnu_pbds::priority_queue<int,greater<int>,pairing_heap_tag> pq;//最快
//----------------------------------------------------------


/*
//----------------------------------------------------------
const int BufferSize = 1 << 16;
char buffer[BufferSize], *_head, *_tail;
inline char Getchar() {
    if (_head == _tail) {
        int l = fread(buffer, 1, BufferSize, stdin);
        _tail = (_head = buffer) + l;
    }
    return *_head++;
 }
inline int read() {
    int x = 0, f = 1;char c = Getchar();
    for (;!isdigit(c);c = Getchar()) if (c == '-') f = -1;
    for (;isdigit(c);c = Getchar()) x = x * 10 + c - '0';
    return x * f;
 }
//----------------------------------------------------------
*/



const int maxn=2e6+9;
const int mod=1e9+7;

int tree[maxn][30],sum[maxn];
int flagg[maxn];
int tot;

int pos[maxn];
bool vis[maxn];

int insert_str(char *str)
{
    int len=strlen(str);
    int rt=0;
    for(int i=0;i<len;i++)
    {
        int id;
        if(str[i]=='?') id=26;
        else if(str[i]=='*') id=27;
        else id=str[i]-'a';
        if(!tree[rt][id]) tree[rt][id]=++tot;
        rt=tree[rt][id];
    }
    flagg[rt]=1;
    return rt;
}
void find_str(char *str,int pos1,int pos2,int len)
{
    if(pos2==len&&flagg[pos1])
    {
        vis[pos1]=1;
    }
    int id=str[pos2]-'a';
    int rt=pos1;
    if(tree[rt][id])
    {
        find_str(str,tree[rt][id],pos2+1,len);
    }
    if(tree[rt][26])
    {
        find_str(str,tree[rt][26],pos2+1,len);
    }
    if(tree[rt][27])
    {
        for(int i=pos2;i<=len;i++) find_str(str,tree[rt][27],i,len);
    }
    return ;
}
void init()
{
    for(int i=0;i<=tot;i++)
    {
        flagg[i]=0;
        for(int j=0;j<30;j++) tree[i][j]=0;
    }
    tot=0;
}
char s[maxn];

int main()
{
    //FAST_IO;

    //freopen("input.txt","r",stdin);

    init();
    int n,m;
    sfi(n);
    sfi(m);
    for(int i=0;i<n;i++)
    {
        sfs(s);
        pos[i]=insert_str(s);
    }
    while(m--)
    {
        bool f=0;
        sfs(s);
        int len=strlen(s);
        find_str(s,0,0,len);
        for(int i=0;i<n;i++)
        {
            if(vis[pos[i]])
            {
                //if(f) printf(" ");
                printf("%d ",i);
                f=1;
                //vis[pos[i]]=0;//会WA，我也不知道为啥
            }
        }
        for(int i=0;i<n;i++)
        {
            if(vis[pos[i]])
            {
                vis[pos[i]]=0;
            }
        }
        if(!f) printf("Not match");
        puts("");
    }
    return 0;
}