BZOJ2351: [BeiJing2011]Matrix

2351: [BeiJing2011]Matrix

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 589  Solved: 171
[Submit][Status]

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%的实际测试数据，M、N ≤ 1000，Q = 10

对于40%的数据，A = 1。

对于80%的数据，A ≤ 10。

对于100%的数据，A ≤ 100。

题解：

因为大小固定的矩阵最多只有100W个，那么我们可以把所有的矩阵hash出来，然后直接查询即可。

代码的实现参考了PoPoQQQ，很有技巧性。

代码：

 #include<cstdio>

 #include<cstdlib>

 #include<cmath>

 #include<cstring>

 #include<algorithm>

 #include<iostream>

 #include<vector>

 #include<map>

 #include<set>

 #include<queue>

 #include<string>

 #define inf 1000000000

 #define maxn 1500

 #define maxm 1000000+5

 #define eps 1e-10

 #define ull unsigned long long

 #define pa pair<int,int>

 #define for0(i,n) for(int i=0;i<=(n);i++)

 #define for1(i,n) for(int i=1;i<=(n);i++)

 #define for2(i,x,y) for(int i=(x);i<=(y);i++)

 #define for3(i,x,y) for(int i=(x);i>=(y);i--)

 #define mod 999983

 using namespace std;

 inline int read()

 {

     int x=,f=;char ch=getchar();

     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}

     while(ch>=''&&ch<=''){x=*x+ch-'';ch=getchar();}

     return x*f;

 }
 int n,m,tot,head[maxm],a,b;
 ull sum[maxn][maxn],power[][maxn];
 struct edge{ull go;int next;}e[maxm];
 const ull base[]={,};
 inline void hash(ull x)
 {
     int y=x%mod;
     e[++tot]=(edge){x,head[y]};head[y]=tot;
 }
 inline bool ask(ull x)
 {
     for(int i=head[x%mod];i;i=e[i].next)if(e[i].go==x)return ;
     return ;
 }

 int main()

 {

     freopen("input.txt","r",stdin);

     freopen("output.txt","w",stdout);

     n=read();m=read();a=read();b=read();
     for1(i,n)for1(j,m){char ch=getchar();while(ch!=''&&ch!='')ch=getchar();sum[i][j]=ch-'';}
     for1(i,n)for1(j,m)sum[i][j]+=sum[i-][j]*base[];
     for1(i,n)for1(j,m)sum[i][j]+=sum[i][j-]*base[];
     power[][]=power[][]=;
     for1(i,maxn-)for0(j,)power[j][i]=power[j][i-]*base[j];
     for2(i,a,n)
      for2(j,b,m)
       hash(sum[i][j]-sum[i-a][j]*power[][a]-sum[i][j-b]*power[][b]+sum[i-a][j-b]*power[][a]*power[][b]);
     int q=read();
     while(q--)
     {
         for1(i,a)for1(j,b){char ch=getchar();while(ch!=''&&ch!='')ch=getchar();sum[i][j]=ch-'';}
         for1(i,a)for1(j,b)sum[i][j]+=sum[i-][j]*base[];
         for1(i,a)for1(j,n)sum[i][j]+=sum[i][j-]*base[];
         if(ask(sum[a][b]))printf("1\n");else printf("0\n");
     }

     return ;

 }