题目描述
给定一个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
题解
二维Hash
因为一维Hash就是一维前缀和,所以二维Hash就是二维前缀和——GXZlegend
事实上的确是这样的,维护矩阵 $(1...n,1...m)$ 的Hash值的方法与二维前缀和类似,利用容斥关系推出。其中,设行列两个base,然后乘上base相加减即可。
由于每次询问的 $a$ 和 $b$ 相同,因此预处理出所有原矩阵的 $a × b$ 的子矩阵的Hash值,存到哈希表中,查询时直接找是否有相等的Hash值即可。
注意:行列base不能相同(不然沿主对角线反转Hash值不变);01串的Hash不能以0和1作为取值,应以‘0‘和‘1‘作为取值。
具体看代码吧。
时间复杂度 $O(nm+qab)$
#include <cstdio> #include <cstring> #include <algorithm> #define N 1010 #define M 12345678 using namespace std; typedef unsigned long long ull; struct data { int head[M] , next[N * N] , tot; ull v[N * N]; inline void insert(ull x) { if(!head[x % M]) head[x % M] = ++tot , v[tot] = x; else { int i; for(i = head[x % M] ; next[i] ; i = next[i]) if(v[i] == x) return; next[i] = ++tot , v[tot] = x; } } inline int count(ull x) { int i; for(i = head[x % M] ; i ; i = next[i]) if(v[i] == x) return 1; return 0; } }mp; ull v[N][N] , w[N][N]; char str[N]; int main() { int n , m , a , b , q , i , j; ull c = 1 , d = 1; scanf("%d%d%d%d" , &n , &m , &a , &b); for(i = 1 ; i <= n ; i ++ ) { scanf("%s" , str + 1); for(j = 1 ; j <= m ; j ++ ) v[i][j] = str[j] + v[i][j - 1] * 233 + v[i - 1][j] * 2333 - v[i - 1][j - 1] * 233 * 2333; } for(i = 1 ; i <= b ; i ++ ) c *= 233; for(i = 1 ; i <= a ; i ++ ) d *= 2333; for(i = a ; i <= n ; i ++ ) for(j = b ; j <= m ; j ++ ) mp.insert(v[i][j] - v[i][j - b] * c - v[i - a][j] * d + v[i - a][j - b] * c * d); scanf("%d" , &q); while(q -- ) { for(i = 1 ; i <= a ; i ++ ) { scanf("%s" , str + 1); for(j = 1 ; j <= b ; j ++ ) w[i][j] = str[j] + w[i][j - 1] * 233 + w[i - 1][j] * 2333 - w[i - 1][j - 1] * 233 * 2333; } printf("%d\n" , mp.count(w[a][b])); } return 0; }