BZOJ2351[BeiJing2011]Matrix——二维hash

题目描述

给定一个M行N列的01矩阵,以及Q个A行B列的01矩阵,你需要求出这Q个矩阵哪些在原矩阵中出现过。
所谓01矩阵,就是矩阵中所有元素不是0就是1。

输入

输入文件的第一行为M、N、A、B,参见题目描述。
接下来M行,每行N个字符,非0即1,描述原矩阵。
接下来一行为你要处理的询问数Q。
接下来Q个矩阵,一共Q*A行,每行B个字符,描述Q个01矩阵。

输出

你需要输出Q行,每行为0或者1,表示这个矩阵是否出现过,0表示没有出现过,1表示出现过。

样例输入

3 3 2 2
111
000
111
3
11
00
11
11
00
11

样例输出

1
0
1

提示

对于100%的实际测试数据,M、N ≤ 1000,Q = 1000

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

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

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

  二维hash,行和列分别用一个base,像求前缀矩阵和一样求前缀矩阵hash。因为A,B是固定的,所以暴力求出所有A*B大小的矩阵hash,用map存一下,对于每个询问O(1)判断。

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cmath>
#include<map>
using namespace std;
int n,m;
char ch[1010];
int A,B;
int Q;
int cnt;
unsigned long long p,q;
map<unsigned long long,int>b;
unsigned long long h[1010][1010];
unsigned long long g[1010][1010];
int main()
{
    scanf("%d%d%d%d",&m,&n,&A,&B);
    for(int i=1;i<=m;i++)
    {
        scanf("%s",ch+1);
        for(int j=1;j<=n;j++)
        {
            h[i][j]=h[i-1][j]*2333+h[i][j-1]*13131-h[i-1][j-1]*2333*13131+ch[j];
        }
    }
    p=1;
    q=1;
    for(int i=1;i<=A;i++)
    {
        p*=2333;
    }
    for(int i=1;i<=B;i++)
    {
        q*=13131;
    }
    for(int i=A;i<=m;i++)
    {
        for(int j=B;j<=n;j++)
        {
            b[h[i][j]-h[i-A][j]*p-h[i][j-B]*q+h[i-A][j-B]*p*q]=++cnt;
        }
    }
    scanf("%d",&Q);
    while(Q--)
    {
        for(int i=1;i<=A;i++)
        {
            scanf("%s",ch+1);
            for(int j=1;j<=B;j++)
            {
                g[i][j]=g[i-1][j]*2333+g[i][j-1]*13131-g[i-1][j-1]*13131*2333+ch[j];
            }
        }
        if(b[g[A][B]]!=0)
        {
            printf("1
");
        }
        else
        {
            printf("0
");
        }
    }
}
原文地址:https://www.cnblogs.com/Khada-Jhin/p/9265242.html