BZOJ 2462: [BeiJing2011]矩阵模板 二维哈希

2462: [BeiJing2011]矩阵模板

Time Limit: 2 Sec  Memory Limit: 128 MB
Submit: 1104  Solved: 516
[Submit][Status][Discuss]

Description

给定一个M行N列的01矩阵，以及Q个A行B列的01矩阵，你需要求出这Q个矩阵哪些在
原矩阵中出现过。 
   所谓01矩阵，就是矩阵中所有元素不是0就是1。 
 

Input

输入文件的第一行为M、N、A、B，参见题目描述。 
接下来M行，每行N个字符，非0即1，描述原矩阵。 
接下来一行为你要处理的询问数Q。 
接下来Q个矩阵，一共Q*A行，每行B个字符，描述Q个01矩阵。 
 

Output

你需要输出Q行，每行为0或者1，表示这个矩阵是否出现过，0表示没有出现过，1表
示出现过。

Sample Input

3 3 2 2
111
000
111
3
11
00
11
11
00
11

Sample Output

1
0
1

HINT

对于100%的数据，N,M<=1000 A,B<=100

谁能体会我内心的崩溃

get 二维哈希

写了个一main到底

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>

using namespace std;

typedef long long ll;
typedef unsigned int ui;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=1010,bas1=131,bas2=193,mod=99999997;

int n,m,a,b,matrix[N][N];

ui mi1[N],mi2[N],h[N][N],ha[N][N];

bool book[mod+10];char s[N];

int main()
{
	n=read();m=read();a=read();b=read();
	register int i,j,Q;ui tmp;
	for(i=1;i<=n;++i){scanf("%s",s+1);for(j=1;j<=m;++j)matrix[i][j]=(s[j]=='1');}
	mi1[0]=mi2[0]=1;
	for(i=1;i<=a;++i)mi1[i]=(mi1[i-1]<<7)+mi1[i-1]+(mi1[i-1]<<1);
	for(i=1;i<=b;++i)mi2[i]=(mi2[i-1]<<7)+mi2[i-1]+(mi2[i-1]<<6);
	for(i=1;i<=n;++i)for(j=1;j<=m;++j)
		h[i][j]=((h[i-1][j]<<7)+h[i-1][j]+(h[i-1][j]<<1)+matrix[i][j]);
	for(i=1;i<=n;++i)for(j=1;j<=m;++j)
		h[i][j]+=((h[i][j-1]<<7)+h[i][j-1]+(h[i][j-1]<<6));
	for(i=a;i<=n;++i)for(j=b;j<=m;++j)
	{
		tmp=h[i][j];
		tmp-=h[i-a][j]*mi1[a];tmp-=h[i][j-b]*mi2[b];
		tmp+=h[i-a][j-b]*mi1[a]*mi2[b];
		tmp=tmp%mod;
		book[tmp%mod]=1;
	}
	Q=read();
	while(Q--)
	{
		for(i=1;i<=a;++i){scanf("%s",s+1);for(j=1;j<=b;++j)matrix[i][j]=(s[j]=='1');}
		memset(ha,0,sizeof(ha));
		for(i=1;i<=a;++i)for(j=1;j<=b;++j)
			ha[i][j]=((ha[i-1][j]<<7)+ha[i-1][j]+(ha[i-1][j]<<1)+matrix[i][j]);
		for(i=1;i<=a;++i)for(j=1;j<=b;++j)
			ha[i][j]+=((ha[i][j-1]<<7)+ha[i][j-1]+(ha[i][j-1]<<6));
		if(book[ha[a][b]%mod])puts("1");
		else puts("0");
	}
	return 0;
}
/*
3 3 2 2 
111
000
111
3
11
00
11
11
00
11

1 
0 
1 
*/