POJ 1204 AC自动机

Posted 2022-05-27 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;