UVA 11019 Matrix Matcher（二维矩阵匹配ac自动机）

题目链接：https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=1960

题意：给出一个n×m的矩阵一，在给出一个x×y的矩阵二，求矩阵二在矩阵一中出现的次数。

思路：对于矩阵二的每行建立Trie，并在单词结尾结点记录走到该结点的为行数c（有多个可开数组记录），利用一个co[r][i]数组记录在矩阵一中以(r, i)为矩阵二的右上角，大小与矩阵二相同的矩阵包含的行数。对矩阵一每行均find()一遍，最后扫描co数组，统计值为x的个数。

注：AC自动机并不是最优解法，时间效率仍是较低，最优解法为二维hash。

代码：

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <vector>

using namespace std;

const int SIZE = 3e2 + 10;
const int N = 1e4 + 10;
const int NN = 1e3 + 10;

struct AC {
    int ch[N][SIZE];
    int sz;
    int f[N];
    bool ed[N];
    int last[N];
    int vt[N][SIZE];
    int vn[N];
    int co[NN][NN];

    int newnode() {
        memset(ch[sz], 0, sizeof(ch[sz]));
        ed[sz] = false;
        f[sz] = 0;
        last[sz] = 0;
        vn[sz] = 0;
        return sz++;
    }

    void init() {
        memset(co, 0, sizeof(co));
        sz = 0;
        newnode();
    }

    void insert(int id, char *s) {
        int u = 0;
        int len = strlen(s);
        for (int i = 0; i < len; i++) {
            int idx = s[i];
            if (!ch[u][idx])
                ch[u][idx] = newnode();
            u = ch[u][idx];
        }
        ed[u] = true;
        vt[u][vn[u]++] = id;
    }

    void getfail() {
        queue<int> q;
        for (int i = 0; i < SIZE; i++)
            if (ch[0][i])
                q.push(ch[0][i]);

        while (!q.empty()) {
            int u = q.front();
            q.pop();
            for (int i = 0; i < SIZE; i++) {
                int v = ch[u][i];
                if (v) {
                    q.push(v);
                    int r = f[u];
                    while (r && !ch[r][i])
                        r = f[r];
                    f[v] = ch[r][i];
                    last[v] = (ed[f[v]] ? f[v] : last[f[v]]);
                }
                else {
                    ch[u][i] = ch[f[u]][i];
                }
            }
        }
    }

    void print(int r, int j, int c) {
        if (j) {
            for (int i = 0; i < vn[j]; i++)
                if (r >= vt[j][i])
                    co[r - vt[j][i]][c]++;
            print(r, last[j], c);
        }
    }

    void find(int r, char *s) {
        int u = 0;
        int len = strlen(s);
        for (int i = 0; i < len; i++) {
            int idx = s[i];
            u = ch[u][idx];
            if (ed[u]) print(r, u, i);
            else if (last[u]) print(r, last[u], i);
        }
    }

}ac;

char Mat[NN][NN];

int main() {

    int t_case;
    scanf("%d", &t_case);
    for (int i_case = 1; i_case <= t_case; i_case++) {
        int n, m, x, y;
        scanf("%d%d", &n, &m);
        for (int i = 1; i <= n; i++)
            scanf("%s", Mat[i]);
        ac.init();
        scanf("%d%d", &x, &y);
        for (int i = 1; i <= x; i++) {
            char str[SIZE];
            scanf("%s", str);
            if (x <= n && y <= m)
                ac.insert(i, str);
        }
        int ans = 0;
        if (x <= n && y <= m) {
            ac.getfail();
            for (int i = 1; i <= n; i++)
                ac.find(i, Mat[i]);

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