POJ1204：Word Puzzles——题解

http://poj.org/problem?id=1204

题目大意：给一个字母表，求一些字符串的开端第一次出现的位置和字符串的方向（字符串可以按照八个方向放在字母表中可匹配的位置）

————————————————————————————————

一定是AC自动机，而且我们不可能对二位字母表AC一下，所以我们要把待匹配串AC一下，然后枚举字母表的起点（不要枚举多了），ACcheck一下就好了，蛮裸的。

为了保证最小序，需要对枚举顺序改一下，具体循环方法我piao了这位大佬的博客。

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
const int L=;
const int C=;
const int M=;
const int W=;
int dx[]={,,,-,,,-,-};
int dy[]={,,-,,-,,,-};
char dir[]={'C','E','G','A','F','D','B','H'};
struct trie{
    int ed;
    int a[];
    int fail;
    int l;
}tree[C*L]={};
char mp[L][C],s[M];
int ans[W][];
bool vis[W];
int cnt=,l,c,w;
inline void insert(int k){
    int now=;
    int len=strlen(s);
    for(int i=;i<len;i++){
        int t=s[i]-'A';
        if(!tree[now].a[t]){
            cnt++;
            tree[now].a[t]=cnt;
        }
        now=tree[now].a[t];
    }
    tree[now].ed=k;
    tree[now].l=len;
    return;
}
int q[C*L];
void getfail(){
    int r=;
    //以下是对第一层的特殊处理
    for(int i=;i<;i++){
        if(tree[].a[i]!=){
            tree[tree[].a[i]].fail=;
        r++;q[r]=tree[].a[i];
        }
    }
    for(int l=;l<=r;l++){
        int u=q[l];
        for(int i=;i<;i++){
            if(tree[u].a[i]!=){
                tree[tree[u].a[i]].fail=tree[tree[u].fail].a[i];
        r++;q[r]=tree[u].a[i];
            }else{
                tree[u].a[i]=tree[tree[u].fail].a[i];
            }
        }
    }
    return;
}
void check(int x,int y,int d){
    int now=;
    while(x>=&&x<l&&y>=&&y<c){
        int t=mp[x][y]-'A';
        now=tree[now].a[t];
        for(int j=now;j;j=tree[j].fail){
        int k=tree[j].ed;
        int len=tree[j].l-;
            if(k&&!vis[k]){
        vis[k]=;
        ans[k][]=x-dx[d]*len;
        ans[k][]=y-dy[d]*len;
        ans[k][]=d;
        }
        }
    x+=dx[d];
    y+=dy[d];
    }
    return;
}
int main(){
    cin>>l>>c>>w;
    for(int i=;i<l;i++){
    cin>>mp[i];
    }
    for(int i=;i<=w;i++){
    cin>>s;
    insert(i);
    }
    getfail();
    for(int i=;i<l;i++)
    for(int j=;j<;j++)
        check(i,,j);
    for(int i=;i<l;i++)
    for(int j=;j<;j++)
        check(i,c-,j);
    for(int i=;i<c;i++)
    for(int j=;j<;j++)
        check(,i,j);
    for(int i=;i<c;i++)
    for(int j=;j<;j++)
        check(l-,i,j);
    for(int i=;i<=w;i++)
    printf("%d %d %c\n",ans[i][],ans[i][],dir[ans[i][]]);
    return ;
}