BZOJ 2351: [BeiJing2011]Matrix 哈希

Time Limit: 20 Sec Memory Limit: 128 MB
Submit: 881 Solved: 271

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。

Source

---

裸哈希过，膜PoPoQQQ，学习了矩阵哈希的方法，真是厉害啊

#include<cstdio>  
#include<cstring>  
#include<iostream>  
#include<algorithm> 
#define BASEI 10016957  
#define BASEII 10016959  
using namespace std;  
struct Hash_Table{  
    unsigned num;  
    int nxt;  
}table[1010*1010];  
int m,n,a,b,q;  
unsigned int sum[1200][1200],powI[1200],powII[1200];  
int head[10100],tail;  
void HASH(unsigned int x){
    int pos=x%10100;
    table[++tail].num=x;
    table[tail].nxt=head[pos];
    head[pos]=tail;
}
bool in_table(unsigned int x){
    int pos=x%10100;
    for(register int i=head[pos];i;i=table[i].nxt)
        if(table[i].num==x) return true;
    return false;
}
int main(){
    scanf("%d%d%d%d",&m,&n,&a,&b);
    for(register int i=1;i<=m;i++)
        for(register int j=1;j<=n;j++)
            scanf("%1d",&sum[i][j]);
    for(register int i=1;i<=m;i++)
        for(register int j=1;j<=n;j++)
            sum[i][j]+=sum[i-1][j]*BASEI;
    for(register int i=1;i<=m;i++)
        for(register int j=1;j<=n;j++)
            sum[i][j]+=sum[i][j-1]*BASEII;
    powI[0]=powII[0]=1;
    for(register int i=1;i<1100;i++){
        powI[i]=powI[i-1]*BASEI;
        powII[i]=powII[i-1]*BASEII;
    }
    for(register int i=a;i<=m;i++)
        for(register int j=b;j<=n;j++){
            unsigned int temp=sum[i][j]-sum[i-a][j]*powI[a]-sum[i][j-b]*powII[b]+sum[i-a][j-b]*powI[a]*powII[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]*BASEI;
        for(register int i=1;i<=a;i++)
            for(register int j=1;j<=b;j++)
                sum[i][j]+=sum[i][j-1]*BASEII;
        unsigned int temp=sum[a][b];
        if(in_table(temp)) printf("1\n");
        else               printf("0\n");
    }
    return 0;
}