【Uva11019】Matrix Matcher【AC自动机】【二维字符串匹配】

http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=33057

二维字符串匹配...大白上的题。

把匹配串的每一行当成一个串insert到AC自动机里，然后用模板串的每一行find。

如果匹配成功，那么在这个矩阵的左上角加一。

最后扫一边矩阵，如果有个地方的权值等于x，那么说明这是个匹配位置。

/* Footprints In The Blood Soaked Snow */
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>

using namespace std;

const int maxnode = 11000, maxn = 1005, maxq = 20000;

int n, m, x, y, q[maxq], cnt[maxn][maxn];

struct _acm {
	int son[maxnode][26], fail[maxnode], acmcnt;
	vector<int> val[maxnode];

	void init() {
		memset(son, 0, sizeof(son)); acmcnt = 0;
		for(int i = 0; i < maxnode; i++) val[i].clear(), fail[i] = 0;
	}

	void insert(char *s, int v) {
		int now = 0;
		for(int i = 0; i < y; i++) {
			int &pos = son[now][s[i] - 'a'];
			if(!pos) pos = ++acmcnt;
			now = pos;
		}
		val[now].push_back(v);
	}

	void getfail() {
		int h = 0, t = 0;
		for(int i = 0; i < 26; i++) if(son[0][i]) q[t++] = son[0][i];
		while(h != t) {
			int u = q[h++];
			for(int i = 0; i < 26; i++)
				if(!son[u][i]) son[u][i] = son[fail[u]][i];
				else fail[q[t++] = son[u][i]] = son[fail[u]][i];
		}
	}

	void find(char *T, int id) {
		int now = 0;
		for(int i = 0; i < m; i++) {
			now = son[now][T[i] - 'a'];
			for(int j = 0; j < val[now].size(); j++) if(id >= val[now][j])
				cnt[id - val[now][j]][i - y + 1]++;
		}
	}
} acm;

char str[maxn][maxn];

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

		scanf("%d%d", &n, &m);
		for(int i = 0; i < n; i++) scanf("%s", str[i]);
		scanf("%d%d", &x, &y);
		for(int i = 0; i < x; i++) {
			char s[maxn]; scanf("%s", s);
			acm.insert(s, i);
		}

		acm.getfail();
		for(int i = 0; i < n; i++)
			acm.find(str[i], i);

		int ans = 0;
		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;
}