BZOJ2351: [BeiJing2011]Matrix

2351: [BeiJing2011]Matrix

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 589  Solved: 171
[Submit][Status]

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 = 10

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

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

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

题解：

因为大小固定的矩阵最多只有100W个，那么我们可以把所有的矩阵hash出来，然后直接查询即可。

代码的实现参考了PoPoQQQ，很有技巧性。

代码：

#include<cstdio>

#include<cstdlib>

#include<cmath>

#include<cstring>

#include<algorithm>

#include<iostream>

#include<vector>

#include<map>

#include<set>

#include<queue>

#include<string>

#define inf 1000000000

#define maxn 1500

#define maxm 1000000+5

#define eps 1e-10

#define ull unsigned long long

#define pa pair<int,int>

#define for0(i,n) for(int i=0;i<=(n);i++)

#define for1(i,n) for(int i=1;i<=(n);i++)

#define for2(i,x,y) for(int i=(x);i<=(y);i++)

#define for3(i,x,y) for(int i=(x);i>=(y);i--)

#define mod 999983

using namespace std;

inline int read()

{

    int x=0,f=1;char ch=getchar();

    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}

    while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}

    return x*f;

}
int n,m,tot,head[maxm],a,b;
ull sum[maxn][maxn],power[2][maxn];
struct edge{ull go;int next;}e[maxm];
const ull base[2]={13131,10007};
inline void hash(ull x)
{
    int y=x%mod;
    e[++tot]=(edge){x,head[y]};head[y]=tot;
}
inline bool ask(ull x)
{
    for(int i=head[x%mod];i;i=e[i].next)if(e[i].go==x)return 1;
    return 0;
}

int main()

{

    freopen("input.txt","r",stdin);

    freopen("output.txt","w",stdout);

    n=read();m=read();a=read();b=read();
    for1(i,n)for1(j,m){char ch=getchar();while(ch!='0'&&ch!='1')ch=getchar();sum[i][j]=ch-'0';}
    for1(i,n)for1(j,m)sum[i][j]+=sum[i-1][j]*base[0];
    for1(i,n)for1(j,m)sum[i][j]+=sum[i][j-1]*base[1];
    power[0][0]=power[1][0]=1;
    for1(i,maxn-1)for0(j,1)power[j][i]=power[j][i-1]*base[j];
    for2(i,a,n)
     for2(j,b,m)
      hash(sum[i][j]-sum[i-a][j]*power[0][a]-sum[i][j-b]*power[1][b]+sum[i-a][j-b]*power[0][a]*power[1][b]);
    int q=read();
    while(q--)
    {
        for1(i,a)for1(j,b){char ch=getchar();while(ch!='0'&&ch!='1')ch=getchar();sum[i][j]=ch-'0';}
        for1(i,a)for1(j,b)sum[i][j]+=sum[i-1][j]*base[0];
        for1(i,a)for1(j,n)sum[i][j]+=sum[i][j-1]*base[1];
        if(ask(sum[a][b]))printf("1\n");else printf("0\n");
    }

    return 0;

}