AC自动机练习题1：地图匹配

AC自动机板子，学习之前要是忘记了就看一下

1465: 【AC自动机】地图匹配

poj1204

时间限制: 1 Sec  内存限制: 256 MB
提交: 78  解决: 46
[提交] [状态] [讨论版] [命题人:admin]

题目描述

【题意】

给出有一个L*C的字符地图，地图的行与列都从0开始编号
然后给出一些字符串，求出这些字符串在字符地图上第一次出现的坐标
输出字符串第一个字母的坐标和字符串的方向
字符串的方向是指字符串的走向
A表示正北，B表示东北，C表示正东，D表示东南，E表示正南，F表示西南，G表示正西，H表示西北
且保证字符串的方向是固定的

【输入格式】
第一行输入L，C，W(0<L,C,W<=1000)
L表示行数，C表示列数，W表示字符串的数量
然后输入L*C的字符矩阵
最后输入W行字符串
【输出格式】
输出W行，每行对应第i个字符串第一个字母的坐标和字符串的方向
【样例输入】
20 20 10
QWSPILAATIRAGRAMYKEI
AGTRCLQAXLPOIJLFVBUQ
TQTKAZXVMRWALEMAPKCW
LIEACNKAZXKPOTPIZCEO
FGKLSTCBTROPICALBLBC
JEWHJEEWSMLPOEKORORA
LUPQWRNJOAAGJKMUSJAE
KRQEIOLOAOQPRTVILCBZ
QOPUCAJSPPOUTMTSLPSF
LPOUYTRFGMMLKIUISXSW
WAHCPOIYTGAKLMNAHBVA
EIAKHPLBGSMCLOGNGJML
LDTIKENVCSWQAZUAOEAL
HOPLPGEJKMNUTIIORMNC
LOIUFTGSQACAXMOPBEIO
QOASDHOPEPNBUYUYOBXB
IONIAELOJHSWASMOUTRK
HPOIYTJPLNAQWDRIBITG
LPOINUYMRTEMPTMLMNBO
PAFCOPLHAVAIANALBPFS
MARGARITA
ALEMA
BARBECUE
TROPICAL
SUPREMA
LOUISIANA
CHEESEHAM
EUROPA
HAVAIANA
CAMPONESA
【样例输出】
0 15 G
2 11 C
7 18 A
4 8 C
16 13 B
4 15 E
10 3 D
5 1 E
19 7 C
11 11 H

首先这个几乎和AC自动机的板子是一样的，只是这个变成了二维的，然后我们现在来看一下这道题要注意的一些小细节

• 首先我们这道题因为一个矩阵当中出现的字母是有很多次的，但是我们要找到最早出现的满足条件的字母，所以我们知道要多增加一个变量来保存我们当前的这个字母有没有被搜索过，然后因为方向的增加，这就意味着我们应该要将每个位置边界位置预处理

• 然后就是说我们的建树要倒着建，为什么呢？因为这样子的话，我们就可以直接在搜索结尾的时候，找到开头的，然后记录下来位置

主要就是这两个细节，剩下的看几乎和AC自动机板子一样的代码

代码的实现

（注释版，我觉得不看注释版都可以理解的代码就尽量不看吧

 /*有一个要注意的，就是横纵左边的起始位置为0，所以我们在计算字符串的时候一定要是开头为0*/
 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<algorithm>
 #include<cmath>
 #include<iostream>
 using namespace std;
 const int dx[]={-,-,,,, , ,-};
 const int dy[]={ , ,,,,-,-,-};
 /*正西 西南 正南 东南 正东 东北 正北 西北*/
 struct trie/*字典树*/
 {
     int s,cnt[],fail,b;
     /*fail就是失败指针，cnt就是孩子，s记录开头所对应的数据，b是记录他有没有被记录过，这里就没有必要初始化了，一百万初始化还是不太好*/
 }tr[];
 struct node/*保存答案的结构体*/
 {
     int x,y,c;/*x，y表示左边，c表示方向*/
 }a[];
 char map[][],s[];
 int tot,list[],n,m,t;
 void clean(int x)/*虽然没有多组数据，但是也要清空，因为我们在建树的时候清空子树可以节省时间*/
 {
     tr[x].b=tr[x].s=tr[x].fail=;
     memset(tr[x].cnt,-,sizeof(tr[x].cnt));
 }
 void build_tree(int id)/*建树的板子*/
 {
     int x=; int len=strlen(s);
     for(int i=len-;i>=;i--)/*倒着建树，方便匹配时记录匹配的终点即为原串的起始点*/
     {
         int y=s[i]-'A';/*没有+1是因为我们从0开始*/
         if(tr[x].cnt[y]==-)
         {
             tr[x].cnt[y]=++tot;
             clean(tot);
         }
         x=tr[x].cnt[y];
     }
     tr[x].s=id;/*记录这个单词开头的编号，因为我们是倒着建树的
     也就是说我们的tot对应的单词是开头的，把他为第几组数据记录下来*/
 }
 void bfs()/*构造失败指针的板子*/
 {
     list[]=; int head=,tail=;
     while(head<=tail)
     {
         int x=list[head];
         for(int i=;i<;i++)
         {
             int son=tr[x].cnt[i];
             if(son==-) continue;
             if(x==) tr[son].fail=;
             else
             {
                 int j=tr[x].fail;
                 while(j!= && tr[j].cnt[i]==-) j=tr[j].fail;
                 tr[son].fail=max(tr[j].cnt[i],);
             }
             list[++tail]=son;
         }
         head++;
     }
 }
 bool check(int x,int y)/*判断一下这个左边能不能进入搜索的范围*/
 {
     if(x< || y< || x>=n || y>=m) return ;
     return ;
 }
 void query(int x,int y,int c)
 {
     int w=;
     while(check(x,y))/*可以进入搜索范围*/
     {
         int j=map[x][y]-'A';/*把他转化为个数*/
         while(w!= && tr[w].cnt[j]==-) w=tr[w].fail;
         if(tr[w].cnt[j]!=-) w=tr[w].cnt[j];/*同样的就是找失败指针来记录*/
         for(int k=w; k!= && !tr[k].b;k=tr[k].fail)
         /*从我们当前当的位置开始，保证他不是根节点，并且没有记录过，然后找他的失败指针*/
         {
             if(tr[k].s)/*找到原串被匹配的起始点*/
             {
                 int l=tr[k].s;
                 a[l].x=x;
                 a[l].y=y;/*记录下来横纵坐标*/
                 a[l].c=(c+)%;/*同种类型的有四种，全部类型有八种*/
             }
             tr[k].b=;/*更新为记录过*/
         }
         x+=dx[c],y+=dy[c];/*按照这个方向往下寻找*/
     }
 }
 int main()
 {
     scanf("%d%d%d",&n,&m,&t);
     clean(); tot=;
     memset(a,-,sizeof(a));/*初始化*/
     for(int i=;i<n;i++) scanf("%s",map[i]);
     /*直接把一行字符输进去，一维的我们是直接map+1，和现在一样*/
     for(int i=;i<=t;i++)
     {
         scanf("%s",s);
         build_tree(i);/*建树*/
     }
     bfs();/*构造失败指针*/
     /*总共有八个方向，0~7，从最上面开始按照顺时针一周*/
     for(int i=;i<n;i++)
     {
         query(i,,); query(i,,); query(i,,);/*右边*/
         query(i,m-,); query(i,m-,); query(i,m-,);/*左边*/
     }
     for(int i=;i<m;i++)
     {
         query(,i,); query(,i,); query(,i,);/*下面*/
         query(n-,i,); query(n-,i,); query(n-,i,);/*上面*/
     }
     /*两重循环加一起起始就是把我们矩阵的外围给围起来的，起始就是说把最外面的有可能走的方向都用AC自动机找一遍*/
     for(int i=;i<=t;i++) printf("%d %d %c\n",a[i].x,a[i].y,a[i].c+'A');
     return ;
 }

Tristan Code 注释版

（非注释版，一个更好锻炼理解代码能力的代码）

 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<algorithm>
 #include<cmath>
 #include<iostream>
 using namespace std;
 const int dx[]={-,-,,,, , ,-};
 const int dy[]={ , ,,,,-,-,-};
 struct trie
 {
     int s,cnt[],fail,b;
 }tr[];
 struct node
 {
     int x,y,c;
 }a[];
 char map[][],s[];
 int tot,list[],n,m,t;
 void clean(int x)
 {
     tr[x].b=tr[x].s=tr[x].fail=;
     memset(tr[x].cnt,-,sizeof(tr[x].cnt));
 }
 void build_tree(int id)
 {
     int x=; int len=strlen(s);
     for(int i=len-;i>=;i--)
     {
         int y=s[i]-'A';
         if(tr[x].cnt[y]==-)
         {
             tr[x].cnt[y]=++tot;
             clean(tot);
         }
         x=tr[x].cnt[y];
     }
     tr[x].s=id;
 }
 void bfs()
 {
     list[]=; int head=,tail=;
     while(head<=tail)
     {
         int x=list[head];
         for(int i=;i<;i++)
         {
             int son=tr[x].cnt[i];
             if(son==-) continue;
             if(x==) tr[son].fail=;
             else
             {
                 int j=tr[x].fail;
                 while(j!= && tr[j].cnt[i]==-) j=tr[j].fail;
                 tr[son].fail=max(tr[j].cnt[i],);
             }
             list[++tail]=son;
         }
         head++;
     }
 }
 bool check(int x,int y)
 {
     if(x< || y< || x>=n || y>=m) return ;
     return ;
 }
 void query(int x,int y,int c)
 {
     int w=;
     while(check(x,y))
     {
         int j=map[x][y]-'A';
         while(w!= && tr[w].cnt[j]==-) w=tr[w].fail;
         if(tr[w].cnt[j]!=-) w=tr[w].cnt[j];
         for(int k=w; k!= && !tr[k].b ;k=tr[k].fail)
         {
             if(tr[k].s)
             {
                 int l=tr[k].s;
                 a[l].x=x;
                 a[l].y=y;
                 a[l].c=(c+)%;
             }
             tr[k].b=;
         }
         x+=dx[c],y+=dy[c];
     }
 }
 int main()
 {
     scanf("%d%d%d",&n,&m,&t);
     clean(); tot=;
     memset(a,-,sizeof(a));
     for(int i=;i<n;i++) scanf("%s",map[i]);
     for(int i=;i<=t;i++)
     {
         scanf("%s",s);
         build_tree(i);
     }
     bfs();
     for(int i=;i<n;i++)
     {
         query(i,,); query(i,,); query(i,,);
         query(i,m-,); query(i,m-,); query(i,m-,);
     }
     for(int i=;i<m;i++)
     {
         query(,i,); query(,i,); query(,i,);
         query(n-,i,); query(n-,i,); query(n-,i,);
     }
     for(int i=;i<=t;i++) printf("%d %d %c\n",a[i].x,a[i].y,a[i].c+'A');
     return ;
 }

Tristan Code 非注释版