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;
}