PKU 1204 Word Puzzles(AC自动机)

题目大意；原题链接

给定一个字符串矩阵和待查找的单词,可以朝8个不同的方向查找,输出待查找单词第一个字母在矩阵中出现的位置和该单词被查到的方向.

A~H代表8个不同的方向,A代表正北方向,其他依次以45度角的方向顺时针增加.

解题思路：

解法一：Trie树暴搜

因为不查询重复单词,所以dfs(int u,int i,int j,int k)函数中当已经查询到单词时,val[u]可以置零做标记.

#include<cstdio>
#include<cstring>
#define maxn 200010
using namespace std;
bool vis[];
int n,m,w,x,y,sz=;//sz得为全局变量
char str[][],word[];
int val[maxn],out[][];
int dir[][]={-,,-,,,,,,,,,-,,-,-,-};//顺时针
struct Trie
{
    int next[];
}trie[maxn];

void insert(char *s,int k)
{
    int u=,len=strlen(s);
    for(int i=;i<len;i++){
        int id=s[i]-'A';
        if(!trie[u].next[id])
            trie[u].next[id]=sz++;
        u=trie[u].next[id];
    }
    val[u]=k;//u为结点编号,k为单词编号
}
void dfs(int u,int i,int j,int k)//k记录方向
{
    if(u==||i<||i>=n||j<||j>=m) return;//该语句之前放在if(val[u])条件之后,WA
    if(val[u]){
        out[val[u]][]=x;
        out[val[u]][]=y;
        out[val[u]][]=k;
        val[u]=;//做标记,因为不查询重复单词
    }
    int id=str[i+dir[k][]][j+dir[k][]]-'A';
    dfs(trie[u].next[id],i+dir[k][],j+dir[k][],k);//必须得朝同一方向搜索
}

int main()
{
    scanf("%d%d%d",&n,&m,&w);
    for(int i=;i<n;i++)
        scanf("%s",str[i]);
    for(int i=;i<=w;i++){
        scanf("%s",word);
        insert(word,i);
        vis[word[]-'A']=;
    }
    for(int i=;i<n;i++){
        for(int j=;j<m;j++){
            if(vis[str[i][j]-'A']){
                for(int k=;k<;k++){
                    x=i,y=j;//记录下最初的位置,从Trie树根开始向下搜索
                    dfs(trie[].next[str[i][j]-'A'],i,j,k);
                }
            }
        }
    }
    for(int i=;i<=w;i++)
        printf("%d %d %c\n",out[i][],out[i][],out[i][]+'A');
    return ;
}

解法二：AC自动机

参考链接:哔哩哔哩算法讲堂

#include<queue>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#define inf 0x3f3f3f3f
using namespace std;
char str[][],word[];
int n,m,w;
int dir[][]={-,,-,,,,,,,,,-,,-,-,-};//顺时针
char ch[]="ABCDEFGH";
int pos[][];
struct TrieNode{
    int id;//记录第几个字符串
    TrieNode *next[],*fail;
    TrieNode(){
        id=,fail=;
        memset(next,,sizeof(next));
    }
}*root;  

void InsertNode(char *s,int id)
{
    int len=strlen(s)-;
    TrieNode *p=root;
    while(len>=){//这里将s数组倒着插入字典树,方便匹配时记录匹配的终点即原串的起始点
        if(!p->next[s[len]-'A'])
            p->next[s[len]-'A']=new TrieNode;
        p=p->next[s[len--]-'A'];
    }
    p->id=id;//id记录单词编号
}
void Build_AC()
{
    TrieNode *p,*next;
    queue<TrieNode* > que;
    que.push(root);
    while (!que.empty()){
        p=que.front();
        que.pop();
        for(int i=;i<;i++){
            if(p->next[i]){
                if(p==root) p->next[i]->fail=root;
                else{
                    TrieNode *temp=p->fail;
                    while(temp){
                         if(temp->next[i]){//temp始终代表next[i]的爸爸,有next[i]这个儿子
                             p->next[i]->fail=temp->next[i];
                            break;//寻找最长后缀
                        }
                        temp=temp->fail;
                    }
                    if(!temp) p->next[i]->fail=root;
                }
                que.push(p->next[i]);
            }
        }
    }
}
void Query(int x,int y,int d,int di)
{
    TrieNode *p=root,*next;
    while(x>=&&y>=&&x<n&&y<m){
        while(p&&!p->next[str[x][y]-'A']) p=p->fail;
        if(!p) p=root;
        else p=p->next[str[x][y]-'A'];
        next=p;
        while(next!=root){
            if(next->id){//记录原串被匹配的起始点
                int k=next->id;
                if(pos[k][]>x||(pos[k][]==x&&pos[k][]>y)){
                    pos[k][]=x;
                    pos[k][]=y;
                    pos[k][]=di;//记录单词整体朝向
                }
            }
            next=next->fail;
        }
        x+=dir[d][];
        y+=dir[d][];
    }
}
void Free(TrieNode *p)
{
    for(int i=;i<;i++){
        if(p->next[i])
            Free(p->next[i]);
    }
    delete p;
} 

int main(){
    scanf("%d%d%d",&n,&m,&w);
    root=new TrieNode;
    for(int i=;i<n;i++)
        scanf("%s",str[i]);
    for(int i=;i<=w;i++){
        scanf("%s",&word);
        InsertNode(word,i);
        pos[i][]=pos[i][]=inf;
    }
    Build_AC();
    for(int i=;i<m;i++){
        Query(,i,,),Query(n-,i,,);
        Query(,i,,),Query(n-,i,,);
        Query(,i,,),Query(n-,i,,);
    }//从矩阵上下左右四条边枚举8个方向
    for(int i=;i<n;i++){
        Query(i,,,),Query(i,m-,,);
        Query(i,,,),Query(i,m-,,);
        Query(i,,,),Query(i,m-,,);
    }
    for(int i=;i<=w;i++)
        printf("%d %d %c\n",pos[i][],pos[i][],ch[pos[i][]]);
    Free(root);
}  

后缀数组倍增解法过一段时间补上