AC自动机 UVa11019

最新推荐文章于 2021-01-18 11:38:28 发布

原创最新推荐文章于 2021-01-18 11:38:28 发布 · 456 阅读

0 ·

CC 4.0 BY-SA版权

AC自动机专栏收录该内容

3 篇文章

订阅专栏

本文探讨使用AC自动机解决二维字符串匹配问题，通过优化算法减少时间复杂度，实现高效匹配。

这题的AC自动机不用维护last数组，找到每个左上角的匹配行数，如果恰好等于x，那么刚好就是一个完整的二维匹配。

但是，不明白为什么一直WA啊，看了别人的代码感觉都差不多啊，我也考虑了模板串多行相同的情况啊，郁闷了。

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;

#define SIG 26
#define N 1005
#define X 105

char t[N][N], s[X];
int cnt[X][X];
int T, ans, n ,m , x, y;

inline int idx(char c)
{
    return c-'a';
}

struct AC_auto {
    int ch[X*X][SIG], val[X*X], trans[X], f[X*X];
    int sz;

    void init(){
        sz=1;
        memset(ch,0,sizeof(ch));
        memset(trans,0,sizeof(trans));
        memset(val,0,sizeof(val));
    }

    void Insert(char *s,int v){
        int u=0, n=strlen(s);
        for (int i=0;i<n;i++){
            int c=idx(s[i]);
            if (!ch[u][c]) {
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        if (val[u]) {
            trans[v]=val[u];
        }
        val[u]=v;
    }

    void getfail(){
        memset(f,0,sizeof(f));
        queue<int> q;
        for (int c=0;c<SIG;c++)
            if (ch[0][c]) {
                f[ch[0][c]]=0;
                q.push(ch[0][c]);
            }
        while (!q.empty()) {
            int r=q.front();
            q.pop();
            for (int c=0;c<SIG;c++) {
                int u=ch[r][c];
                if (!u) {
                    ch[r][c]= ch[f[r]][c];
                    continue;
                }
                q.push(u);
                int v=f[r];
                while (v && !ch[v][c]) v=f[v];
                f[u] = ch[v][c];
            }
        }
    }

    void Find(char *t,int row,int col){
        int u=0, n=strlen(t);
        for (int i=0;i<n;i++) {
            int c=idx(t[i]);
            u=ch[u][c];
            if (val[u]){
                int tmp=val[u];
                while (tmp){
                    if (row-tmp+1>=0 && i-col+1>=0)
                        cnt[row-tmp+1][i-col+1]++;
                    tmp=trans[tmp];
                }
            }
        }
    }
}ac;

int main()
{
    scanf("%d",&T);
    while (T--) {
        ac.init();
        memset(cnt,0,sizeof(cnt));
        ans=0;

        scanf("%d%d",&n,&m);
        for (int i=0;i<n;i++)
            scanf("%s",t[i]);
        scanf("%d%d",&x,&y);
        for (int i=1;i<=x;i++) {
            scanf("%s",s);
            ac.Insert(s,i);
        }
        ac.getfail();
        for (int i=0;i<n;i++)
            ac.Find(t[i],i,y);

        for (int i=0;i<n;i++)
            for (int j=0;j<m;j++)
                if (cnt[i][j]==x) ans++;
        printf("%d\n",ans);
    }

    return 0;
}