【POJ1204】Word Puzzles-AC自动机

测试地址：Word Puzzles

题目大意：在一个填满大写字母的N*M棋盘内找到询问的W个单词，输出出现点坐标和方向（方向从“向上为A”顺时针依次定义，如向右上为B，向右为C......）。

做法：对所有待查字符串构建AC自动机，然后对每个方向查找一遍棋盘即可求出答案。注意下标从0开始。

以下是本人代码（由于我一开始没注意，所以处理过程中下标是从1开始的，所以到输出的时候坐标减去了1）：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
int n,m,q,rt,tot=0,len;
int fail[1000010],ch[1000010][30],id[1000010]={0},fa[1000010],ans[1010][3];
int dep[1000010];
char puz[1010][1010],s[1010];

void newnode(int &v)
{
  v=++tot;
  for(int i=0;i<=25;i++) ch[tot][i]=0;
}

void insert(int &v,int c,int p)
{
  if (c>len) return;
  if (!v) newnode(v);
  int x=s[c]-'A';
  insert(ch[v][x],c+1,p);
  if (ch[v][x]) fa[ch[v][x]]=v;
  if (c==len) id[v]=p;
}

void calcfail()
{
  queue<int> q;
  dep[rt]=0;
  for(int i=0;i<=25;i++)
    if (ch[rt][i]) dep[ch[rt][i]]=1,q.push(ch[rt][i]),fail[ch[rt][i]]=rt;
  while(!q.empty())
  {
    int v=q.front();q.pop();
	for(int i=0;i<=25;i++)
	  if (ch[v][i])
	  {
	    int x=fail[v];
		dep[ch[v][i]]=dep[v]+1;
		while(!ch[x][i]&&fail[x]) x=fail[x];
		if (ch[x][i]) fail[ch[v][i]]=ch[x][i];
		else fail[ch[v][i]]=rt;
	    q.push(ch[v][i]);
	  }
  }
}

void search(int x,int y,int dx,int dy,int mode,int v)
{
  if (id[v])
  {
    int i=v;
	while(fail[i])
	{
	  if (id[i]) ans[id[i]][0]=x-dep[i]*dx,ans[id[i]][1]=y-dep[i]*dy,ans[id[i]][2]=mode;
      i=fail[i];
	}
  }
  if (x<1||x>n||y<1||y>m) return;
  int xx=puz[x][y]-'A';
  while(!ch[v][xx]&&fail[v]) v=fail[v];
  if (ch[v][xx]) search(x+dx,y+dy,dx,dy,mode,ch[v][xx]);
  else search(x+dx,y+dy,dx,dy,mode,v);
}

int main()
{
  scanf("%d%d%d",&n,&m,&q);
  for(int i=1;i<=n;i++)
    scanf("%s",&puz[i][1]);
  newnode(rt);
  fail[rt]=0;
  for(int i=1;i<=q;i++)
  {
    scanf("%s",s);
	len=strlen(s);
	insert(rt,0,i);
  }
  
  calcfail();
  for(int i=1;i<=m;i++)
  {
    search(n,i,-1,0,0,rt);
	search(n,i,-1,1,1,rt);
	search(n,i,-1,-1,7,rt);
    search(1,i,1,0,4,rt);
	search(1,i,1,1,3,rt);
	search(1,i,1,-1,5,rt);
  }
  for(int i=1;i<=n;i++)
  {
    search(i,1,0,1,2,rt);
	search(i,1,1,1,3,rt);
	search(i,1,-1,1,1,rt);
	search(i,m,0,-1,6,rt);
	search(i,m,1,-1,5,rt);
	search(i,m,-1,-1,7,rt);
  }
  
  for(int i=1;i<=q;i++)
    printf("%d %d %c\n",ans[i][0]-1,ans[i][1]-1,ans[i][2]+'A');
  
  return 0;
}