poj 1204 字符串处理 多串匹配自动机

#include<cstdio>
#include<cstring>
#include<stdlib.h>
#define N 1010
struct node
{
       node *next[26];
       node *fail;
       int count,index;
       node()
       {
            count=0,index=0,fail=NULL;
            for(int i=0;i<26;i++)next[i]=NULL;
       }
}*root,*q[501000];
int head,tail;
int n,m,w,sum,x[N],y[N];
char map[N][N];
char str[N],dir[N];
int len[N];
int h[8]={-1,-1,0,1,1,1,0,-1};
int c[8]={0,1,1,1,0,-1,-1,-1};
void insert(int ii)
{
     int i=0,index;
     node *temp=root;
     len[ii]=strlen(str);
     while(str[i])
     {
           index=str[i]-'A';
           if(temp->next[index]==NULL)temp->next[index]=new node();
           temp=temp->next[index];
           i++;
     }
     temp->count=1;
     temp->index=ii;
}
void build_ac()
{
     node *temp,*p;
     head=tail=0;
     q[head++]=root;
     while(head!=tail)
     {
           temp=q[tail++];
           for(int i=0;i<26;i++)
           {
               if(temp->next[i]!=NULL)
               {
                   if(temp==root)temp->next[i]->fail=root;
                   else
                   {
                       p=temp->fail;
                       while(p!=NULL)
                       {
                             if(p->next[i]!=NULL)
                             {
                                temp->next[i]->fail=p->next[i];
                                break;
                             }
                             p=p->fail;
                       }
                       if(p==NULL)temp->next[i]->fail=root;
                   }
                   q[head++]=temp->next[i];
               }
           }
     }
}
bool Is_OK(int x,int y)
{
     if(x>=n||x<0||y<0||y>=m)
        return false;
     return true;
}
void query(int i,int j,int k)
{
     int cx=i,cy=j,ind;
     node *temp=root,*p;
     for(;Is_OK(cx,cy);cx+=h[k],cy+=c[k])
     {
          ind=map[cx][cy]-'A';
          printf("%d\n",ind);
          while(temp->next[ind]==NULL&&temp!=root)
          {
                temp=temp->fail;
                
          }
          temp=temp->next[ind];
          p=(temp==NULL)?root:temp;
          while(p!=root)
          {
                if(p->count)
                {
                   sum++;
                   p->count=0;
                   x[p->index]=cx-(h[k]*len[p->index]);
                   y[p->index]=cy-(c[k]*len[p->index]);
                   dir[p->index]='A'+k;
                }
                p=p->fail;
          }
     }
}
void solution()
{
     for(int i=0;i<n&&sum<w;i++)
     {
         query(i,0,1);query(i,0,2);query(i,0,3);
         query(i,m-1,5);query(i,m-1,6);query(i,m-1,7);
     }
     for(int i=0;i<m&&sum<w;i++)
     {
         query(0,i,3);query(0,i,4);query(0,i,5);
         query(n-1,i,0);query(n-1,i,1);query(n-1,i,7);
     }
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&w))
    {
         root=new node();
         for(int i=0;i<n;i++)
             scanf("%s",map[i]);
         sum=0;
         for(int i=0;i<w;i++)
         {
             scanf("%s",str);
             insert(i+1);
         }
         build_ac();
         solution();
         for(int i=1;i<=w;i++)
             printf("%d %d %c\n",x[i],y[i],dir[i]);

    }
    return 0;
}

题意：

告诉你一篇文章，也就是一个nxm的字符矩阵，然后给你w个单词，问这些单词在文章中出现的位置，输出坐标，然后现在有八个方向。

同时还要在输出的坐标位置后面输出相应的匹配方向。

为了节省时间，我们肯定首先要建立一个字典树。

然后做法有两种，一种是直接从文章中的每一个点出发从八个方向，在字典树当中比较。

一种是建立ac自动机，构建字典树当中的fail指针，从文章的边界出发，将所有的能匹配的方向都扫一遍就ok了。。。

字典树做法：

#include<cstdio>
#include<cstring>
#define N 1010
struct node
{
       node *next[26];
       int count,index;
       node()
       {
           count=0,index=0;
           for(int i=0;i<26;i++)next[i]=NULL;
       }
}*root;
char map[N][N];
char str[N];
int n,m,w,sum;
int x[N],y[N];
char dir[N];
int h[8]={-1,-1,0,1,1,1,0,-1};
int c[8]={0,1,1,1,0,-1,-1,-1};
void insert(int ii)
{
     node *p=root;
     int i=0;
     while(str[i])
     {
           int ind=str[i]-'A';
           if(p->next[ind]==NULL)p->next[ind]=new node();
           p=p->next[ind];
           i++;
     }
     p->count=1;
     p->index=ii;
}
bool Is_OK(int x,int y)
{
     if(x>=0&&x<n&&y>=0&&y<m)
     return true;
     return false;
}
void query(int i,int j,int k)
{
      int cx=i,cy=j,index;
      node *temp=root,*p;
      for(;Is_OK(cx,cy);cx+=h[k],cy+=c[k])
      {
          index=map[cx][cy]-'A';
          if(temp->next[index]==NULL)
             return ;
          temp=temp->next[index];
          if(temp->count)
          {
             sum++;
             x[temp->index]=i;
             y[temp->index]=j;
             dir[temp->index]='A'+k;
             temp->count=0;
          }
      }
}
void solution()
{
     for(int i=0;i<n;i++)
     {
         for(int j=0;j<m;j++)
         {
            for(int k=0;k<8;k++)
            {
               query(i,j,k);
               if(sum==w)break;
            }
            if(sum==w)break;
         }
         if(sum==w)break;
     }
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&w))
    {
        root=new node();
        for(int i=0;i<n;i++)
            scanf("%s",map[i]);
        for(int i=0;i<w;i++)
        {
            scanf("%s",str);
            insert(i+1);
        }
        sum=0;
        solution();
        for(int i=1;i<=w;i++)
            printf("%d %d %c\n",x[i],y[i],dir[i]);
    }
    return 0;
}

ac自动机: