poj 1204 AC自动机

AC自动机的经典题

可以使用一个小技巧，将要查找的单词反过来建自动机，这样便于记录首字母的位置。

#include <iostream>
using namespace std;
const int maxn=1005,maxl=200000;
const int dx[9]={0,-1,-1,0,1,1,1,0,-1};
const int dy[9]={0,0,1,1,1,0,-1,-1,-1};


struct node
{
       node* ch[26];
       node* fail;
       int no;            
};
node* root=new node;
char s[maxn],a[maxn][maxn];
int gx[maxn],gy[maxn],ansx[maxn],ansy[maxn],ansd[maxn];
int i,j,k,dir,n,m,kk;


void ins(char* s,int x)
{
     node* now=root;
     int i,j,k,len=strlen(s);
     for (i=len-1;i>=0;i--)
     {
         k=s[i]-'A';
         if (now->ch[k]!=0) 
            now=now->ch[k];
         else 
         {
              now->ch[k]=new node;
              now=now->ch[k];
              for (int j=0;j<26;j++)
                  now->ch[j]=0;
              now->no=-1;now->fail=root;
         }
     }
     now->no=x;
}


node* q[maxl];
void build()
{
     int i,h=0,r=1;
     node *now,*ff;
     root->fail=0;
     q[++h]=root;
     while (h<=r)
     {
           now=q[h++];
           for (i=0;i<26;i++)
               if (now->ch[i]!=0)
               {
                  q[++r]=now->ch[i];
                  for (ff=now->fail;ff!=0;ff=ff->fail)
                      if (ff->ch[i]!=0)
                      {
                         now->ch[i]->fail=ff->ch[i];
                         break;
                      }
               }
     }
}


void match(char* s)
{
     int i,j,k,len=strlen(s);
     node* now=root,*tmp;
     for (i=0;i<len;i++)
     {
         k=s[i]-'A';
         while (now->ch[k]==0 && now!=root)
               now=now->fail;
         if (now->ch[k]==0) continue;
         now=now->ch[k];
         tmp=now;
         while (tmp!=0 && tmp->no!=-1)
         {
               ansx[tmp->no]=gx[i];
               ansy[tmp->no]=gy[i];
               ansd[tmp->no]=dir;
               tmp->no=-1;
               tmp=tmp->fail;
         }
     }
}         


void make(int x,int y,int dir)
{
     int len=0;
     while (x>=0 && x<n && y>=0 && y<m)  
     {
           s[len]=a[x][y];
           gx[len]=x;gy[len]=y;
           x+=dx[dir];y+=dy[dir];
           len++;
     }
}        
       
int main()
{
    freopen("pin.txt","r",stdin);
    freopen("pou.txt","w",stdout);
    cin >> n >> m >> kk;
    for (i=0;i<26;i++)
        root->ch[i]=0;
    root->no=-1;
    getchar();
    for (i=0;i<n;i++)
    {
        for (j=0;j<m;j++)
            a[i][j]=getchar();
        getchar();
    }
    for (i=0;i<kk;i++)
    {
        cin >> s;
        ins(s,i);
    }
    build();
    for (i=0;i<n;i++)
        for (j=0;j<m;j++)
            if (i==0 || i==n-1 || j==0 || j==m-1)
            for (k=1;k<=8;k++)
            {
                if (k==4) make(i,j,8);
                else make(i,j,(k+4)%8);
                dir=k;
                match(s);
            }
    for (i=0;i<kk;i++)
        printf("%d %d %c\n",ansx[i],ansy[i],ansd[i]+'A'-1);
    return 0;
}