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;}