1204--Word Puzzles

原创于 2021-04-22 22:23:31 发布 · 624 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法 #字符串

本文介绍了一种使用AC自动机算法来解决给定L*C字符矩阵中寻找w个单词首次出现的位置和方向的问题。通过构建字典树并从矩阵边界开始搜索，作者提供了C++代码实现和详细步骤。

1204–Word Puzzles

链接

题意：
给一个L*C的字符矩阵，然后w个单词，要求每个单词在字符矩阵中首次出现的位置和方向。

分析：
将代求单词建立字典树，利用AC自动机。从字符矩阵中遍历各个方向和起点查找，起点只需从矩阵四条边开始就行，因为从边界出发总能覆盖到每条边。

code:

#include <iostream>
#include <string>
#include <set>
#include<queue>
#include<stdio.h>
#include<string.h>
using namespace std;
typedef unsigned long long ULL;
const int Max=1e6+5;
char c[1005][1005];
int t[Max][26];
int l[Max];
int dx[8]={-1,-1,0,1,1,1,0,-1};
int dy[8]={0,1,1,1,0,-1,-1,-1};
int cnt=0;
int n,m;
int e[Max];
string str[1020];
int ans1[1020],ans2[1020],ans3[1020];
void fuck(string s,int id)
{
    int p=0;
    for(int i=0;i<(int)s.size();i++)
    {
        int a=s[i]-'A';
        if(!t[p][a]) t[p][a]=++cnt;
        p=t[p][a];
    }
    e[p]=id;
}
void build()
{
    queue<int>Q;
    for(int i=0;i<26;i++)
    {
        if(t[0][i]) Q.push(t[0][i]);
    }
    while(Q.size())
    {
        int u=Q.front();
        Q.pop();
        for(int i=0;i<26;i++)
        {
            if(t[u][i]){
                l[t[u][i]]=t[l[u]][i];
                Q.push(t[u][i]);
            }
            else{
               t[u][i]=t[l[u]][i];
            }
        }
    }
}
void query(int x,int y,int r)
{
    int u=0;
    while(x>=0&&y>=0&&x<n&&y<m)
    {

        u=t[u][c[x][y]-'A'];
        for(int j=u;j;j=l[j])
        {
            if(e[j])  {
                ans1[e[j]]=x-((int)str[e[j]].size()-1)*dx[r];
                ans2[e[j]]=y-((int)str[e[j]].size()-1)*dy[r];
                ans3[e[j]]=r;
                }
        }
        x+=dx[r];
        y+=dy[r];
    }
    return ;
}

int main()
{
    int t;
    cin>>n>>m>>t;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            cin>>c[i][j];
        }
    }
   for(int i=1;i<=t;i++)
   {
       cin>>str[i];
       fuck(str[i],i);
   }
   build();
   for(int i=0;i<n;i++)
   {
       for(int j=0;j<8;j++)
       {
           query(0,i,j);
       }
   }
   for(int i=0;i<n;i++)
   {
       for(int j=0;j<8;j++)
       {
           query(i,0,j);
           query(i,m-1,j);

       }
   }
   for(int i=0;i<n;i++)
   {
       for(int j=0;j<8;j++)
       {
           query(n-1,i,j);
       }
   }
   for(int i=1;i<=t;i++)
   {
       cout<<ans1[i]<<" "<<ans2[i]<<" "<<char(ans3[i]+'A')<<endl;
   }
   return 0;
}