poj 1204 AC自动机

利用要匹配的串建立tire树，然后枚举四周的字母分别向8个方向做多模式串匹配。

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<iostream>
using namespace std;
const int nMax=2000000;
const int S_NUM=26;
int gx[8]={-1,-1,0,1,1,1,0,-1};
int gy[8]={0,1,1,1,0,-1,-1,-1};
char h[10]="ABCDEFGH";
struct Tree
{
    int index,flag;
    Tree *fail,*next[S_NUM];
}t[nMax],*Q[nMax];
int cnt,L,C,W,b[1005][3],len[1005];
Tree *root;
char str[3005],a[1005][1005];
int get(char c)
{
    return c-'A';
}
int ok(int x,int y)
{
    return (x>=0&&x<L&&y>=0&&y<C);
}
Tree* new_node()
{
    Tree *p=&t[cnt];
    p->index=cnt++;
    p->flag=-1;
    memset(p->next,0,sizeof(p->next));
    return p;
}
void insert(char s[],int f)
{
    Tree *p=root;
    for(int i=0;s[i];i++)
    {
        int index=get(s[i]);
        if(p->next[index]==NULL)
            p->next[index]=new_node();
        p=p->next[index];
    }
    p->flag=f;
}
void build_ac()
{
    int f=0,r=-1;
    Q[++r]=root;
    while(f<=r)
    {
        Tree *p=Q[f++];
        for(int i=0;i<S_NUM;i++)
        {
            if(p->next[i]!=NULL)
            {
                if(p==root) p->next[i]->fail=root;
                else
                {
                    p->next[i]->fail=p->fail->next[i];
                }
                Q[++r]=p->next[i];
            }
            else
            {
                if(p==root) p->next[i]=root;
                else p->next[i]=p->fail->next[i];
            }
        }
    }
}
void find(int x,int y,int d)
{
    Tree *p=root;
    while(ok(x,y))
    {
        int index=get(a[x][y]);
        p=p->next[index];
        Tree *tmp=p;
        while(tmp!=root)
        {
            if((tmp->flag)!=-1)
            {
                int f=tmp->flag;
                b[f][0]=x-(len[f]-1)*gx[d];b[f][1]=y-(len[f]-1)*gy[d];b[f][2]=d;
            }
            tmp=tmp->fail;
        }
        x+=gx[d];y+=gy[d];
    }
}
int main()
{
//    freopen("test.txt","r",stdin);
    scanf("%d%d%d",&L,&C,&W);
    for(int i=0;i<L;i++)
        scanf("%s",a[i]);
    cnt=0; root=new_node();
    for(int i=0;i<W;i++)
    {
        scanf("%s",str);
        len[i]=strlen(str);
        insert(str,i);
    }
    build_ac();
    for(int i=0;i<C;i++)
    {
        find(0,i,4);find(0,i,3);find(0,i,5);
        find(L-1,i,0);find(L-1,i,1);find(L-1,i,7);
    }
    for(int i=0;i<L;i++)
    {
        find(i,0,2);find(i,0,1);find(i,0,3);
        find(i,C-1,6);find(i,C-1,5);find(i,C-1,7);
    }
    for(int i=0;i<W;i++)
        printf("%d %d %c\n",b[i][0],b[i][1],h[b[i][2]]);
    return 0;

}