【AC自动机/二维哈希】UVA11019 Matrix Matcher

这道题目首先想到的就是二维hash！

那就先写一发

 

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1005;
const int b1=19260817;
const int b2=233;
char s1[maxn][maxn],s2[maxn][maxn];
ll h1[maxn][maxn],h2[maxn][maxn],p1[maxn*maxn],p2[maxn*maxn];
void calc(char s[][maxn],ll h[][maxn],int n,int m)
{
	for(int i=0;i<n;i++)
		for(int j=0;j<m;j++)
			h[i+1][j+1]=h[i][j+1]*b1+h[i+1][j]*b2-h[i][j]*b1*b2+s[i][j];
}
ll has(ll h[][maxn],int x1,int y1,int x2,int y2)
{
	return h[x2+1][y2+1]-h[x1][y2+1]*p1[x2-x1+1]-h[x2+1][y1]*p2[y2-y1+1]+h[x1][y1]*p1[x2-x1+1]*p2[y2-y1+1];
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("a.out","w",stdout);
	int t; scanf("%d",&t);
	int n,m,nn,mm;
	p1[0]=p2[0]=1;
	for(int i=1;i<=maxn*maxn;i++) p1[i]=p1[i-1]*b1,p2[i]=p2[i-1]*b2;
	while(t--)
	{
		scanf("%d%d",&n,&m);
		for(int i=0;i<n;i++) scanf("%s",s1[i]);
		calc(s1,h1,n,m);	
		scanf("%d%d",&nn,&mm);
		for(int i=0;i<nn;i++) scanf("%s",s2[i]);
		calc(s2,h2,nn,mm);
		int ans=0;
		for(int i=0;i<n;i++)
			if(i+nn-1<n)
				for(int j=0;j<m;j++)
					if(j+mm-1<m && has(h1,i,j,i+nn-1,j+mm-1)==h2[nn][mm]) ans++;
		printf("%d\n",ans);
	}
	return 0;
}

 

第二种方法：因为要用n*m的矩阵去匹配x*y的矩阵，所以我们可以对x*y进行建立自动机，可以把这个矩阵看成x个字符串，每个长度为y

对于n*m的每一行，在AC自动机里走一遍，记录下每一个字符时节点的位置，这时就得到了每个位置为终点，可以匹配到哪里

如果连续若干行每个位置分别匹配到1-x的某个位置，就解决了问题

对于矩阵C上的某一位置C[i][j]表示：从第i行第j列的字符开始向右匹配，能和编号为C[i][j]的字符串匹配成功。

 

 

代码

#include<bits/stdc++.h>
using namespace std;
int nr,nc,mr,mc;
const int maxn=1005; 
const int maxx=1e4+5;
char s[maxn][maxn],s2[maxn][maxn]; 
int cnt[maxn][maxn],num[maxx];
int L,root,val[maxx],tr[maxx][27];
int getnewnode()
{
	for(int i=0;i<26;i++)
		tr[L][i]=0;
	val[L++]=0;
	return L-1;
}
void init()
{
	memset(cnt,0,sizeof(cnt));
	memset(num,0,sizeof(num));
	L=0;
	root=getnewnode();
}
void insert(char *s,int id)
{
	int now=root,len=strlen(s);
	for(int i=0;i<len;i++)
	{
		int dig=s[i]-'a';
		if(!tr[now][dig]) tr[now][dig]=getnewnode();
		now=tr[now][dig];
	}
	if(val[now]) num[id+1]=val[now];
	val[now]=id+1;
}
int fail[maxx],last[maxx];
void build()
{
	fail[root]=root;
	queue <int> q;
	for(int i=0;i<26;i++)
	{
		int u=tr[0][i];
		if(u)
		{
			fail[u]=0;
			last[u]=0;
			q.push(u);
		}
	}
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(int i=0;i<26;i++)
		{
			int now=tr[u][i];
			if(!now)
			{
				tr[u][i]=tr[fail[u]][i];
				continue;
			}
			q.push(now);
			int v=fail[u];
			while(v && !tr[v][i]) v=fail[v];
			fail[now]=tr[v][i];
			last[now]=val[fail[now]]?fail[now]:last[fail[now]];			
		}
	}
}
void print(int x,int y,int j)
{
	if(j)
	{
		if(x-val[j]+1>=0)
			cnt[x-val[j]+1][y]++;
		int t=val[j];
		while(num[t])
		{
			t=num[t];
			if(x-t+1>=0)
				cnt[x-t+1][y]++;
		}
		print(x,y,last[j]);
	}
}
void find(char *s,int r)
{
	int now=root,len=strlen(s);
	for(int i=0;i<len;i++)
	{
		int dig=s[i]-'a';
		now=tr[now][dig];
		if(val[now]) print(r,i,now);
		else if(last[now]) print(r,i,last[now]);
	}
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("a.out","w",stdout);
	int t; scanf("%d",&t); 
	while(t--)
	{
		init();
		scanf("%d%d",&nr,&nc);
		for(int i=0;i<nr;i++) scanf("%s",s[i]);
		scanf("%d%d",&mr,&mc);
		for(int i=0;i<mr;i++)
		{
			scanf("%s",s2[i]);
			insert(s2[i],i);
		} 
		build();
		int ans=0;
		for(int i=0;i<nr;i++) find(s[i],i);
		for(int i=0;i<nr;i++)
			for(int j=0;j<nc;j++)
				if(cnt[i][j]==mr)
					ans++;
		printf("%d\n",ans);
	}
	return 0;
}