POJ 1204 AC自动机

Posted 2020-07-07 Dan__ge

点击打开链接

题意：给个L*C的字符串矩阵，W个询问，对每个询问输出这个串第一次出现的位置及方向，共有8个方向，用A~H表示

思路：用AC自动机进行快速匹配，细节处理特别多，不看题解的话应该会WA很多次，还有一个处理的非常巧妙地地方，就是我们要输出第一次出现的位置，而AC自动机不能回溯，所以将串反着构造进字典树里，真是神犇，然后方向设置时也反着就可以了

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=500010;
const int N=26;
struct node{
    node *fail;
    node *next[N];
    int num;
    node(){
        fail=NULL;
        num=0;
        memset(next,NULL,sizeof(next));
    }
}*q[maxn];
node *root;
void insert_Trie(char *str,node *root,int id){
    node *p=root;
    int i=strlen(str)-1;
    while(i>=0){
        int id=str[i]-'A';
        if(p->next[id]==NULL) p->next[id]=new node();
        p=p->next[id];i--;
    }
    p->num=id;
}
void build_ac(node *root){
    root->fail=NULL;
    int head=0,tail=0;
    q[head++]=root;
    while(head!=tail){
        node *temp=q[tail++];
        node *p=NULL;
        for(int i=0;i<N;i++){
            if(temp->next[i]!=NULL){
                if(temp==root) temp->next[i]->fail=root;
                else{
                    p=temp->fail;
                    while(p!=NULL){
                        if(p->next[i]!=NULL){
                            temp->next[i]->fail=p->next[i];
                            break;
                        }
                        p=p->fail;
                    }
                    if(p==NULL) temp->next[i]->fail=root;
                }
                q[head++]=temp->next[i];
            }
        }
    }
}
//上，右上，右，右下，下，左下，左，左上；
int dir[8][2]={{-1,0},{-1,1},{0,1},{1,1},{1,0},{1,-1},{0,-1},{-1,-1}};
int ans[1010][10];
int L,C,W;
char str[1010][1010];
char str1[10010];
void query(int x,int y,int pos,int id){
    node *p=root,*temp;
    while(x>=0&&y>=0&&x<L&&y<C){
        int idid=str[x][y]-'A';
        while(p->next[idid]==NULL&&p!=root) p=p->fail;
        p=p->next[idid];
        p=(p==NULL)?root:p;
        temp=p;
        while(temp!=root){
            if(temp->num){
                int ttt=temp->num;
                if(ans[ttt][0]>x||(ans[ttt][0]==x&&ans[ttt][1]>y)){
                    ans[ttt][0]=x;ans[ttt][1]=y;ans[ttt][2]=id;
                }
            }
            temp=temp->fail;
        }
        x=x+dir[pos][0];
        y=y+dir[pos][1];
    }
}
void del(node *p){
     if(p==NULL)return ;
     for(int i=0;i<26;i++)del(p->next[i]);
     delete p;
}
int main(){
    while(scanf("%d%d%d",&L,&C,&W)!=-1){
        root=new node();
        for(int i=0;i<L;i++) scanf("%s",str[i]);
        for(int i=1;i<=W;i++){
            scanf("%s",str1);
            insert_Trie(str1,root,i);
            ans[i][0]=ans[i][1]=inf;
        }
        build_ac(root);
        for(int i=0;i<L;i++){
            query(i,0,2,6);query(i,C-1,6,2);
            query(i,0,1,5);query(i,C-1,5,1);
            query(i,0,3,7);query(i,C-1,7,3);
        }
        for(int i=0;i<C;i++){
            query(0,i,4,0);query(L-1,i,0,4);
            query(0,i,5,1);query(L-1,i,1,5);
            query(0,i,3,7);query(L-1,i,7,3);
        }
        for(int i=1;i<=W;i++){
            char ch=ans[i][2]+'A';
            printf("%d %d %c\n",ans[i][0],ans[i][1],ch);
        }
        del(root);
    }
    return 0;
}