BZOJ2462矩陣模板

2462: [BeiJing2011]矩陣模板
Time Limit: 2 Sec Memory Limit: 128 MB
Submit: 721 Solved: 320
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%的數據，A < = 100。
Source
Day4
見BZOJ2351題解。。
附上本蒟蒻的代碼：

#include<cstdio>
#include<iostream>
using namespace std;
#define MAXN 1001
#define mod 77171
#define base1 9804799
#define base2 9983543
int n,m,A,B,q,hash[mod],tot;
unsigned int a[MAXN][MAXN],b[MAXN][MAXN],pow1[MAXN],pow2[MAXN],Hash;
struct kx
{
    unsigned int num;
    kx *next;
}s[MAXN*MAXN];

int read()
{
    int w=0,c=1; char ch=getchar();
    while (ch<'0' || ch>'9')
      {
        if (ch=='-') c=-1;
        ch=getchar();
      }
    while (ch>='0' && ch<='9')
      w=w*10+ch-'0',ch=getchar();
    return w*c;
}

void insert(unsigned int x)
{
    int t=x%mod;
    s[++tot].num=x,s[tot].next=&s[hash[t]],hash[t]=tot;
}

bool find(unsigned int x)
{
    int t=x%mod;
    for (kx *i=&s[hash[t]];i;i=i->next)
      if (i->num==x) return 1;
    return 0;
}

int main()
{
    int i,j;
    m=read(),n=read(),A=read(),B=read();
    pow1[0]=pow2[0]=1;
    for (i=1;i<=m;i++)
      for (j=1;j<=n;j++)
        scanf("%1d",&a[i][j]);
    for (i=1;i<=max(n,m);i++) pow1[i]=pow1[i-1]*base1,pow2[i]=pow2[i-1]*base2;
    for (i=1;i<=m;i++)
      for (j=1;j<=n;j++)
        a[i][j]+=a[i-1][j]*base1;
    for (i=1;i<=m;i++)
      for (j=1;j<=n;j++)
        a[i][j]+=a[i][j-1]*base2;
    for (i=A;i<=m;i++)
      for (j=B;j<=n;j++)
        {
            Hash=a[i][j],Hash-=a[i-A][j]*pow1[A],Hash-=a[i][j-B]*pow2[B],Hash+=a[i-A][j-B]*pow1[A]*pow2[B];
            insert(Hash);
        }
    q=read();
    while (q--)
      {
        for (i=1;i<=A;i++)
          for (j=1;j<=B;j++)
            scanf("%1d",&b[i][j]);
        for (i=1;i<=A;i++)
          for (j=1;j<=B;j++)
            b[i][j]+=b[i-1][j]*base1;
        for (i=1;i<=A;i++)
          for (j=1;j<=B;j++)
            b[i][j]+=b[i][j-1]*base2;
        if (find(b[A][B])) printf("1\n");
        else printf("0\n");
      }
    return 0;
}