bzoj2462: [BeiJing2011]矩阵模板

2462: [BeiJing2011]矩阵模板

Time Limit: 2 Sec  Memory Limit: 128 MB
Submit: 1961  Solved: 826
[Submit][Status][Discuss]

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%的数据，N,M<=1000 A,B<=100

 

二维哈希的模板题

二维哈希就是分行分列哈希，比如说先哈希每列，那么就相当于变成了一维数组，再当作一维数组哈希

这个时候数组里Hash[ i ][ j ]存的就是从左上角（1，1）到（i，j）的哈希值

利用二维前缀和，计算出每个a*b的矩阵的哈希值，存入数组

最后二分比较即可

对，用了map然后 t 到自闭（太菜了）

据说数据很水，输出十个1就可以A，不过模板题还是存个模板方便以后比较好

代码：

#include <bits/stdc++.h>

using namespace std;
typedef unsigned long long ll;
const int maxn = 1e3 + 100;
const int seed1 = 31;
const int seed2 = 131;
const int mod = 1e9 + 7;
ll f1[maxn], f2[maxn];
ll Hash[maxn][maxn];
char s[maxn][maxn], ss[maxn][maxn];
int data[maxn * maxn];

void init() { //系数
    f1[0] = f2[0] = 1;
    for (int i = 1; i < 120; i++) {
        f1[i] = f1[i - 1] * seed1;
        f2[i] = f2[i - 1] * seed2;
    }
}

ll CalHash(int a, int b) {
    ll ans = 0;
    for (int i = 1; i <= a; i++) {
        ll temp = 0;
        for (int j = 1; j <= b; j++) {
            temp = temp * seed1 + (ss[i][j] - '0');
        }
        ans = ans * seed2 + temp;
    }
    return ans % mod;
}

int main() {
    int n, m, a, b, k = 0;
    scanf("%d%d%d%d", &n, &m, &a, &b);
    init();
    for (int i = 1; i <= n; i++) scanf("%s", s[i] + 1);
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++)
            Hash[i][j] += Hash[i][j - 1] * seed1 + (s[i][j] - '0'); //对每一行哈希 前缀和
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++)
            Hash[i][j] += Hash[i - 1][j] * seed2; //列哈希 前缀和
    }

    for (int i = a; i <= n; i++) {
        for (int j = b; j <= m; j++) {  //计算原矩阵中每个a*b矩阵的哈希值
            ll v = Hash[i][j] - Hash[i - a][j] * f2[a] - Hash[i][j - b] * f1[b] + Hash[i - a][j - b] * f1[b] * f2[a];
            data[k++] = (int) (v % mod);
        }
    }
    sort(data, data + k);
    int q;
    scanf("%d", &q);
    while (q--) {
        for (int i = 1; i <= a; i++) scanf("%s", ss[i] + 1);
        int ha = (int) CalHash(a, b) % mod;
        if (data[lower_bound(data, data + k, ha) - data] == ha) printf("1\n");
        else printf("0\n");
    }
    return 0;
}