poj1204

一开始做这道题没什么头绪，那时只会写Trie，ac自动机不怎么懂。后来在discuss里看到要用AC自动机做。

先把要查询的字符串添加进Trie图里，建完自动机后再从矩形四周八个方向依次查询，结果放在数组里。

#include<cstdio>
#include<string.h>
#include<iostream>
#include<queue>
using namespace std;
const int maxn=1001;
char g[maxn][maxn],T[maxn][maxn],str[maxn];
int ch[maxn*maxn][26],val[maxn*maxn],f[maxn*maxn],last[maxn*maxn],vis[maxn];
int L,C,W,top,ans[maxn][4];
int dir[][2]={{-1,0},{-1,1},{0,1},{1,1},{1,0},{1,-1},{0,-1},{-1,-1}};
void init()
{
    top=1;
    memset(ch[0],0,sizeof(ch[0]));
    memset(ans,0,sizeof(ans));
}
int idx(char c)
{
    return c-'A';
}
void insert(char *s,int op)
{
    int u=0,n=strlen(s),c;
    for(int i=0;i<n;i++)
    {
        c=idx(s[i]);
        if(ch[u][c]==0)
        {
            memset(ch[top],0,sizeof(ch[top]));
            val[top]=0;
            ch[u][c]=top++;
        }
        u=ch[u][c];
    }
    val[u]=op+1;
    //printf("val=%d\n",val[u]);
}

void getfail()
{
    queue<int> q;
    f[0]=0;
    for(int c=0;c<26;c++)
    {
        int u=ch[0][c];
        if(u){f[u]=0;q.push(u);last[u]=0;}
    }
    while(!q.empty())
    {
        int r=q.front();q.pop();
        for(int c=0;c<26;c++)
        {
            int u=ch[r][c];
            if(!u) continue;
            q.push(u);
            int v=f[r];
            while(v&&!ch[v][c]) v=f[v];
            f[u]=ch[v][c];
            last[u]=val[f[u]]?f[u]:last[f[u]];
        }
    }
}

bool ok(int x,int y)
{
    return (x>=0&&x<L&&y>=0&&y<C);
}

void print(int j,int pos,int x,int y,int d)
{
    //printf("j=%d d=%d x=%d y=%d ",j,d,x,y);
    int n=strlen(T[j]);
    int i;
    //printf("%d")
    n=pos-n;
    //printf("n=%d pos=%d ",n,pos);
    int nx=x,ny=y;
    for(i=0;i<=n;i++)
    {
        nx+=dir[d][0];
        ny+=dir[d][1];
    }
    //printf("nx=%d ny=%d\n",nx,ny);
    ans[j][0]=nx;
    ans[j][1]=ny;
    ans[j][2]=d;
}
void find(char *T,int x,int y,int d)
{
    int n=strlen(T);
    int i,j=0;
    for(i=0;i<n;i++)
    {
        int c=T[i]-'A';
        while(j&&!ch[j][c]) j=f[j];
        j=ch[j][c];
        if(val[j]) print(val[j]-1,i,x,y,d);
        else if(last[j]) print(val[last[j]]-1,i,x,y,d);
    }
}
void make(int x,int y,int d)
{
    int nx=x,ny=y,j=0;
    while(ok(nx,ny))
    {
        str[j++]=g[nx][ny];
        nx+=dir[d][0];
        ny+=dir[d][1];
    }
    str[j]='\0';
    find(str,x,y,d);
}
void work()
{
    getfail();
    memset(vis,0,sizeof(vis));
    int i,j;
    for(i=0;i<L;i++)
    {
        for(j=0;j<8;j++)
        {
            make(i,0,j);
            make(i,C-1,j);
        }
    }
    for(i=0;i<C;i++)
    {
        for(j=0;j<8;j++)
        {
            make(0,i,j);
            make(L-1,i,j);
        }
    }
}


int main()
{
    //freopen("test.txt","r",stdin);
    scanf("%d%d%d",&L,&C,&W);
    int i,j;
    getchar();
    init();
    for(i=0;i<L;i++)
    scanf("%s",g[i]);
    for(i=0;i<W;i++)
    {
        scanf("%s",T[i]);
        insert(T[i],i);
    }
    work();
    for(i=0;i<W;i++)
    {
        printf("%d %d %c\n",ans[i][0],ans[i][1],ans[i][2]+'A');
    }
    return 0;
}