【bzoj2351】[BeiJing2011]Matrix hash表+双hash

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

写个hash表练习
双hash是我闲的蛋疼
这个A不了bzoj2462，因为卡常了…模多了就是不好啊

所以那个题直接输出1就行了233

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

typedef long long LL;
const int SZ = 2010;
const int SZ2 = 1000010;
const int base1 = 13331;
const int base2 = 1333331;
const int mod1 = 1000003;
const int mod2 = 998244353;

char a[SZ][SZ];

struct pair{
    int x,y;
}h_sum[SZ][SZ],mi1[SZ],mi2[SZ],hh[SZ][SZ];

struct hashlist{
    int x;
    int cnt;
    hashlist *nxt;
}*h[SZ2],T[SZ2];

int Tcnt = 0;

hashlist* newnode(int x)
{
    hashlist *k = T + (Tcnt ++);
    k -> x = x;
    k -> cnt = 1;
    k -> nxt = NULL;
    return k;
}

void insert(int pos,int x)
{
    pos = (pos % mod1 + mod1) % mod1;
    x = (x % mod2 + mod2) % mod2;
    if(!h[pos]) h[pos] = newnode(x);
    else for(hashlist *p = h[pos];p;p = p -> nxt)
    {
        if(p -> x == x)
        {
            p -> cnt ++;
            break;
        }
        if(!p -> nxt) 
        {
            p -> nxt = newnode(x);
            break;
        }
    }
}

bool find(int x,int y)
{
    x = (x % mod1 + mod1) % mod1;
    y = (y % mod2 + mod2) % mod2;
    for(hashlist *p = h[x];p;p = p -> nxt)
        if(p -> x == y) return true;
    return false;
}

char s[SZ][SZ];

int main()
{
    int n,m,r,c;
    scanf("%d%d%d%d",&n,&m,&r,&c);
    for(int i = 1;i <= n;i ++)
        scanf("%s",a[i] + 1);
    mi1[0].x = mi1[0].y = 1;
    mi2[0].x = mi2[0].y = 1;
    for(int i = 1;i <= 1010;i ++)
    {
        mi1[i].x = (LL)mi1[i - 1].x * base1 % mod1;
        mi2[i].x = (LL)mi2[i - 1].x * base2 % mod1;
        mi1[i].y = (LL)mi1[i - 1].y * base1 % mod2;
        mi2[i].y = (LL)mi2[i - 1].y * base2 % mod2;
    }
    for(int i = 1;i <= n;i ++)
        for(int j = 1;j <= m;j ++)
        {
            h_sum[i][j].x = ((LL)h_sum[i][j - 1].x * base1 % mod1 + a[i][j] - '0' + 1) % mod1;
            h_sum[i][j].y = ((LL)h_sum[i][j - 1].y * base1 % mod2 + a[i][j] - '0' + 1) % mod2;
        }
    for(int i = 1;i <= n;i ++)
        for(int j = 1;j <= m;j ++)
        {
            h_sum[i][j].x = ((LL)h_sum[i - 1][j].x * base2 % mod1 + h_sum[i][j].x) % mod1;
            h_sum[i][j].y = ((LL)h_sum[i - 1][j].y * base2 % mod2 + h_sum[i][j].y) % mod2;
        }

    for(int i = 1;i <= n;i ++)
        for(int j = 1;j <= m;j ++)
        {
            int x = h_sum[i][j].x; 
            x = (x - (LL)h_sum[i - r][j].x * mi2[r].x % mod1) % mod1; 
            x = (x - (LL)h_sum[i][j - c].x * mi1[c].x % mod1) % mod1; 
            x = (x + (LL)h_sum[i - r][j - c].x * mi1[c].x % mod1 * mi2[r].x % mod1) % mod1; 
            int y = h_sum[i][j].y; 
            y = (y - (LL)h_sum[i - r][j].y * mi2[r].y % mod2) % mod2; 
            y = (y - (LL)h_sum[i][j - c].y * mi1[c].y % mod2) % mod2; 
            y = (y + (LL)h_sum[i - r][j - c].y * mi1[c].y % mod2 * mi2[r].y % mod2) % mod2;             
            insert(x,y);
        }   
    int q;
    scanf("%d",&q);
    while(q --)
    {
        for(int i = 1;i <= r;i ++)
            scanf("%s",s[i] + 1);
        for(int i = 1;i <= r;i ++)
            for(int j = 1;j <= c;j ++)
            {
                hh[i][j].x = ((LL)hh[i][j - 1].x * base1 % mod1 + s[i][j] - '0' + 1) % mod1;
                hh[i][j].y = ((LL)hh[i][j - 1].y * base1 % mod2 + s[i][j] - '0' + 1) % mod2;
            }
        for(int i = 1;i <= r;i ++)
            for(int j = 1;j <= c;j ++)
            {
                hh[i][j].x = ((LL)hh[i - 1][j].x * base2 % mod1 + hh[i][j].x) % mod1;
                hh[i][j].y = ((LL)hh[i - 1][j].y * base2 % mod2 + hh[i][j].y) % mod2;
            }       
        puts(find(hh[r][c].x,hh[r][c].y) ? "1" : "0");
    }
    return 0;
}