POJ1204 Word Puzzles（AC自动机）

给一个L*C字符矩阵和W个字符串，问那些字符串出现在矩阵的位置，横竖斜八个向。

就是个多模式匹配的问题，直接AC自动机搞了，枚举字符矩阵八个方向的所有字符串构成主串，然后在W个模式串构造的AC自动机上跑。

另外，temp指针的那个找遗漏后缀的过程执行时标记一下，下一次再到这个结点就不需要再进行一次temp的过程，这样的时间复杂度就是O(W个模式串总长+LC)。

一开始还想8个方向分别计算坐标= =写第二个方向懒得写了，然后就忽然想到可以一开始构造主串时就存坐标。。最后代码很是挺长的。。

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define MAXN 111111
int tn,ch[MAXN][26],flag[MAXN],fail[MAXN],len[MAXN];
void insert(char *s,int k,int l){
    int x=0;
    for(int i=0; s[i]; ++i){
        int y=s[i]-'A';
        if(ch[x][y]==0) ch[x][y]=++tn;
        x=ch[x][y];
    }
    flag[x]=k;
    len[x]=l;
}
void getFail(){
    memset(fail,0,sizeof(fail));
    queue<int> que;
    for(int i=0; i<26; ++i){
        if(ch[0][i]) que.push(ch[0][i]);
    }
    while(!que.empty()){
        int x=que.front(); que.pop();
        for(int i=0; i<26; ++i){
            if(ch[x][i]) que.push(ch[x][i]),fail[ch[x][i]]=ch[fail[x]][i];
            else ch[x][i]=ch[fail[x]][i]; 
        }
    }
}

int n,m;
int ansx[1111],ansy[1111],anso[1111];
char str[1111]; int sx[1111],sy[1111];
void match(char o){
    int x=0;
    for(int i=0; str[i]; ++i){
        int y=str[i]-'A';
        x=ch[x][y];
        int tmp=x;
        while(flag[tmp] && tmp){
            if(flag[tmp]>0){
                int &idx=flag[tmp];
                ansx[idx]=sx[i-len[tmp]+1];
                ansy[idx]=sy[i-len[tmp]+1];
                anso[idx]=o;
                idx=-1;
            }else if(flag[tmp]==-1) break;
            tmp=fail[tmp];
        }
    }
}
char map[1111][11111];
int main(){
    int w;
    scanf("%d%d%d",&n,&m,&w);
    for(int i=0; i<n; ++i) scanf("%s",map[i]);
    for(int i=1; i<=w; ++i){
        scanf("%s",str);
        insert(str,i,strlen(str));
    }
    getFail();
    int sn;
    //zero
    for(int i=0; i<m; ++i){
        for(int j=0; j<n; ++j){
            sx[n-j-1]=j; sy[n-j-1]=i;
            str[n-j-1]=map[j][i];
        }
        str[n]=0;
        match('A');
    }
    //one
    for(int i=0; i<n; ++i){
        int x=i,y=0,sn=0;
        while(x>=0 && y<m){
            sx[sn]=x; sy[sn]=y;
            str[sn++]=map[x][y];
            --x; ++y;
        }
        str[sn]=0;
        match('B');
    }
    for(int i=1; i<m; ++i){
        int x=n-1,y=i,sn=0;
        while(x>=0 && y<m){
            sx[sn]=x; sy[sn]=y;
            str[sn++]=map[x][y];
            --x; ++y;
        }
        str[sn]=0;
        match('B');
    }
    //two
    for(int i=0; i<n; ++i){
        for(int j=0; j<m; ++j){
            sx[j]=i; sy[j]=j;
            str[j]=map[i][j];
        }
        str[m]=0;
        match('C');
    }
    //three
    for(int i=0; i<m; ++i){
        int x=0,y=i,sn=0;
        while(x<n && y<m){
            sx[sn]=x; sy[sn]=y;
            str[sn++]=map[x][y];
            ++x; ++y;
        }
        str[sn]=0;
        match('D');
    }
    for(int i=1; i<n; ++i){
        int x=i,y=0,sn=0;
        while(x<n && y<m){
            sx[sn]=x; sy[sn]=y;
            str[sn++]=map[x][y];
            ++x; ++y;
        }
        str[sn]=0;
        match('D');
    }
    //four
    for(int i=0; i<m; ++i){
        for(int j=0; j<n; ++j){
            sx[j]=j; sy[j]=i;
            str[j]=map[j][i];
        }
        str[n]=0;
        match('E');
    }
    //five
    for(int i=0; i<m; ++i){
        int x=0,y=i,sn=0;
        while(x<n && y>=0){
            sx[sn]=x; sy[sn]=y;
            str[sn++]=map[x][y];
            ++x; --y;
        }
        str[sn]=0;
        match('F');
    }
    for(int i=1; i<n; ++i){
        int x=i,y=m-1,sn=0;
        while(x<n && y>=0){
            sx[sn]=x; sy[sn]=y;
            str[sn++]=map[x][y];
            ++x; --y;
        }
        str[sn]=0;
        match('F');
    }
    //six
    for(int i=0; i<n; ++i){
        for(int j=0; j<m; ++j){
            sx[m-j-1]=i; sy[m-j-1]=j;
            str[m-j-1]=map[i][j];
        }
        str[m]=0;
        match('G');
    }
    //seven
    for(int i=0; i<m; ++i){
        int x=n-1,y=i,sn=0;
        while(x>=0 && y>=0){
            sx[sn]=x; sy[sn]=y;
            str[sn++]=map[x][y];
            --x; --y;
        }
        str[sn]=0;
        match('H');
    }
    for(int i=0; i<n-1; ++i){
        int x=i,y=m-1,sn=0;
        while(x>=0 && y>=0){
            sx[sn]=x; sy[sn]=y;
            str[sn++]=map[x][y];
            --x; --y;
        }
        str[sn]=0;
        match('H');
    }
    for(int i=1; i<=w; ++i){
        printf("%d %d %c
",ansx[i],ansy[i],anso[i]);
    }
    return 0;
}