BZOJ 2351: [BeiJing2011]Matrix

2351: [BeiJing2011]Matrix

Time Limit: 20 Sec Memory Limit: 128 MB
Submit: 903 Solved: 281
[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%的实际测试数据，M、N ≤ 1000，Q = 1000

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

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

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

分析：
考虑hash，就可以实现O（1）查询某个矩阵是否出现过。
但当原矩阵进行hash时，可以利用模数可减的性质，先求前缀和，再减去这个矩阵最上一横排之前的那些行，和左边不包含在这个矩阵中的这些列，再加上左上角被减了2次的小矩阵。
hash果然是靠信仰。
第一次写矩阵HASH的我不要脸地抄了神犇的代码，包括base
自己取的base WA掉了 难过
以及为什么unsigned long long 不行？。。

#include<cstdio>#include<cstring>#include<algorithm>#include<iostream>using namespace std;const int N = 1010;const int base1 = 10016957;const int base2 = 10016959;const int mod = 10007;int m,n,a,b,q;  unsigned int sum[N][N],power1[N],power2[N];struct node{    int pre;    unsigned int w;}edge[N * N];inline int Max(int a,int b){    return a>b?a:b;}int num=0;int head[N*N];void add(int from,unsigned int temp){    num++;    edge[num].pre=head[from];    edge[num].w=temp;    head[from]=num;}inline void Hash(unsigned int x){    int pos=x%mod;    add(pos,x);}inline bool check(unsigned int temp){    int pos=temp%mod;    for(register int i=head[pos];i;i=edge[i].pre)        if(temp==edge[i].w) return true;    return false;} int main(){    scanf("%d%d%d%d",&n,&m,&a,&b);    for(register int i=1;i<=n;i++)       for(register int j=1;j<=m;j++)          scanf("%1d",&sum[i][j]);    for(register int i=1;i<=n;i++)        for(register int j=1;j<=m;j++)           sum[i][j]+=sum[i-1][j]*base1;    for(register int i=1;i<=n;i++)        for(register int j=1;j<=m;j++)           sum[i][j]+=sum[i][j-1]*base2;    power1[0]=power2[0]=1;    for(register int i=1;i<=max(n,m);i++){       power1[i]=power1[i-1]*base1;       power2[i]=power2[i-1]*base2;    }    for(register int i=a;i<=n;i++){        for(register int j=b;j<=m;j++){            unsigned int temp = sum[i][j]            -sum[i-a][j]*power1[a]            -sum[i][j-b]*power2[b]            +sum[i-a][j-b]*power1[a]*power2[b];            Hash(temp);        }    }    scanf("%d",&q);    while(q--){        for(register int i=1;i<=a;i++)           for(register int j=1;j<=b;j++)              scanf("%1d",&sum[i][j]);        for(register int i=1;i<=a;i++)           for(register int j=1;j<=b;j++)              sum[i][j]+=sum[i-1][j]*base1;        for(register int i=1;i<=a;i++)           for(register int j=1;j<=b;j++)              sum[i][j]+=sum[i][j-1]*base2;        unsigned int temp = sum[a][b];        if(check(temp)) printf("1\n");        else printf("0\n");    }    return 0;}