UVa11019 Matrix Matcher

        题意:矩阵匹配,有一个大矩阵,一个小矩阵,问小矩阵在大矩阵里面出现了多少次。

        思路:AC自动机。简单分析一下暴力肯定是不行的,因为是白书AC自动机的例题,想都没想就套AC自动机了。因为矩阵匹配是由x个行匹配构成的,首先给小矩阵每一行建字典树,造AC自动机,然后对大矩阵每一行去匹配。开一个整型矩阵计数,当出现行匹配时,给可能的矩阵匹配的右上角那个元素+1。到最后扫一遍计数矩阵,如果值等于小矩阵行数x,则说明出现了一个矩阵匹配。

        需要注意的是,小矩阵可能出现相同的行。所以出现一个行匹配,可能同时匹配了不止一行,需要稍作处理。


#include <iostream>             
#include <stdio.h>             
#include <cmath>             
#include <algorithm>             
#include <iomanip>             
#include <cstdlib>             
#include <string>             
#include <memory.h>             
#include <vector>             
#include <queue>             
#include <stack>             
#include <map>           
#include <set>           
#include <ctype.h>           
#define INF 1<<30         
#define ll long long         
#define max3(a,b,c) max(a,max(b,c))         

using namespace std;

#define maxnode 1010

int n,m;
int x,y;
char mat[1010][1010];
char pat[110][110];

//ac
int sz;
int ch[11000][26];
vector<int> val[11000];
int next[11000];
int cnt[1010][1010];

void init(){
	sz=1;
	memset(ch[0],0,sizeof(ch[0]));
	memset(val,0,sizeof(val));
	memset(next,0,sizeof(next));
	memset(cnt,0,sizeof(cnt));
}

void insert(char* str,int w){
	//length is y
	int u=0;
	for(int i=0;i<y;i++){
		if( !ch[u][str[i]-'a'] ){
			ch[u][str[i]-'a']=sz;
			memset(ch[sz],0,sizeof(ch[sz]));
			sz++;
		}
		u=ch[u][str[i]-'a'];
	}
	val[u].push_back(w);
}

void build_ac(){
	queue<int> que; que.push(0);
	while(!que.empty()){
		int u=que.front(); que.pop();
		for(int i=0;i<26;i++){
			int v=ch[u][i];
			if(!v)continue;
			
			if(!u){
				next[v]=0;
				que.push(v);
			}else{
				int t=next[u];
				while(t&& !ch[t][i]){
					t=next[t];
				}
				if(ch[t][i])t=ch[t][i];
				next[v]=t;
				que.push(v);
			}
		}
	}
}


void query(char* text,int nn){
	//lenght is m
	int u=0;
	for(int i=0;i<m;i++){
		int c=text[i]-'a';
		while(u&& !ch[u][c] ){
			u=next[u];
		}
		if(ch[u][c])u=ch[u][c];
		int siz=val[u].size();
		for(int j=0;j<siz;j++){
			if(nn-val[u][j]+1>=0)cnt[nn-val[u][j]+1][i]++;
		}
		
	}
}
//end ac

int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		init();
		//big mat
		scanf("%d%d",&n,&m);
		for(int i=0;i<n;i++){
			scanf("%s",mat[i]);
		}
		
		//pattern
		scanf("%d%d",&x,&y);
		for(int i=0;i<x;i++){
			scanf("%s",pat[i]);
			insert(pat[i],i+1);
		}
		build_ac();
		
		for(int i=0;i<n;i++){
			query(mat[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++;
			}
		}
		
		cout<<ans<<endl;
	}
	return 0;
}



你可能感兴趣的:(AC自动机)