BZOJ 2351 Matrix （Hash）

2351: [BeiJing2011]Matrix

Time Limit: 20 Sec Memory Limit: 128 MB
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。

思路：
首先将原矩阵哈希，将所有a*b的子矩阵的哈希值插入哈希表，然后对于每个矩阵哈希之后去哈希表中查找。

#include <cstdio>  #include <cstring>  #include <iostream>  #include <algorithm> #define LL long long #define M 1010  #define base1 10000007  #define base2 10000013 #define mod 10007  using namespace std;  struct EE{    unsigned num;      int next;}ste[M*M];  unsigned int mix[M][M], pow1[M], pow2[M];  int m, n, a, b, q, idc;  int head[20000];  void hs(unsigned int x){//保存hash值，链表避免冲突     int pos = x % mod;    ste[++idc].num = x;      ste[idc].next = head[pos];      head[pos] = idc;  }  bool find(unsigned int x){      int pos = x % mod;      for(int i=head[pos]; i; i=ste[i].next)          if(ste[i].num == x)              return true;      return false;  }  int main(){     scanf("%d%d%d%d", &m, &n, &a, &b);    for(int i=1; i<=m; i++)          for(int j=1; j<=n; j++)              scanf("%1d", &mix[i][j]);      for(int i=1; i<=m; i++)          for(int j=1; j<=n; j++)              mix[i][j] += mix[i-1][j] * base1;      for(int i=1; i<=m; i++)          for(int j=1; j<=n; j++)              mix[i][j] += mix[i][j-1] * base2;      pow1[0] = pow2[0] = 1;      for(int i=1; i<1005; i++){               pow1[i] = pow1[i-1] * base1;          pow2[i] = pow2[i-1] * base2;      }     for(int i=a; i<=m; i++)//原矩阵中所有ab矩阵的hash          for(int j=b; j<=n; j++){              unsigned int temp;            temp = mix[i][j] - mix[i-a][j]*pow1[a] - mix[i][j-b]*pow2[b] + mix[i-a][j-b]*pow1[a]*pow2[b];              hs( temp );          }      scanf("%d", &q);    while( q-- ){          for(int i=1; i<=a; i++)              for(int j=1; j<=b; j++)                  scanf("%1d", &mix[i][j]);          for(int i=1; i<=a; i++)//计算当前矩阵的hash             for(int j=1; j<=b; j++)                 mix[i][j] += mix[i-1][j]*base1;          for(int i=1; i<=a; i++)              for(int j=1; j<=b; j++)                 mix[i][j] += mix[i][j-1]*base2;          unsigned int temp = mix[a][b];        if( find(temp) ) puts("1");        else puts("0");      }    return 0;}