https://vjudge.net/contest/247319#problem/C
题目大意就是给出一个n * m的矩阵,矩阵中只有一些*或者0,n <= 1000, m <= 1000
然后有t (t <= 100)个询问,每次询问给出一个p * q的矩阵,p,q是提前固定的数值。
问这些询问中是大矩阵的子矩阵的有几个
如果暴力查询,时间复杂度为t*n*m*p*q。
用哈希优化一下,我们每次询问的时候把询问的矩阵用哈希处理成一个数字,然后把大矩阵的每一个p*q的子矩阵都处理成一个数字,这样做的时间复杂度为n*m(我自己估算的值)。
我们已经知道了如何用哈希处理一个字符串,就是把字符串看成一个b进制的数字,每一个数位上的值由字符串本身决定。一个字符串的哈希值已经求出来之后,它后移一位的字符串的哈希值也很好求,只要乘一个b,再把前面的那一位减掉,并把后面那一位加上就可以了。(这是针对固定长度的字符串)
然后如何求一个矩阵字符串的哈希值呢?很简单,把每一行的哈希值求出来,再把求出来的哈希值看成一个字符串,取另外一个值b2来计算这个新字符串的哈希值即可。
unsigned long long 无符号位整型,溢出时会自动取模。另外,在计算字符串哈希的时候每一位上的数值不要取0,因为如果取零的话会不好区分字符串的长度。
multiset,包含在头文件set当中,与set的区别是可以包含重复元素。
#include <iostream>
#include <set>
using namespace std;
typedef unsigned long long ull;
const ull b1=9997, b2=1000007;
char s[1005], h[100];
ull a[1005][1005], b[1005][1005], tmp[1005][1005], ha[1005][1005];
int n, m, t, p, q;
multiset<ull> unseen;
void cal(ull a[1005][1005], int n, int m) //计算矩阵a的所有p*q的子矩阵的哈希值
{
ull e=0;
ull t1=1;
for (int j=0; j<q; j++) t1*=b1;
for (int i = 0; i < n; i++) //计算每一行从任意位置开始的长为q的字符串的哈希值
{
ull e=0;
for (int j = 0; j < q; j++) e=e*b1+a[i][j];
for (int j = 0; j+q <= m; j++)
{
tmp[i][j]=e;
if (j+q<m) e=e*b1-t1*a[i][j]+a[i][j+q];
}
}
ull t2=1;
for (int i = 0; i < p; i++) t2 *= b2;
for (int j = 0; j+q <= m; j++){ //计算从任意一个位置开始的p*q的矩阵的哈希值
ull e=0;
for (int i = 0; i < p; i++) e=e*b2+tmp[i][j];
for (int i = 0; i+p <= n; i++){
ha[i][j]=e;
if (i+p<n) e=e*b2-t2*tmp[i][j]+tmp[i+p][j];
}
}
}
int main()
{
int u=0;
while (scanf("%d%d%d%d%d", &n, &m, &t, &p, &q)!=EOF)
{
if (!n && !m && !t && !p && !q) break;
u++;
for (int i = 0; i < n; i++)
{
scanf("%s", &s);
for (int j = 0; j < m; j++)
{
a[i][j]=s[j];
}
}
unseen.clear();
for (int i = 0; i < t; i++)
{
for (int j = 0; j < p; j++)
{
scanf("%s", &h);
for (int k = 0; k < q; k++)
{
b[j][k]=h[k];
}
}
cal(b, p, q);
unseen.insert(ha[0][0]); //把所有的询问的矩阵的哈希值存在multiset中
}
cal(a, n, m); //计算大矩阵中的所有p*q的子矩阵的哈希值
for (int i = 0; i+p <= n; i++) //枚举大矩阵中所有的p*q的子矩阵的哈希值
{
for (int j = 0; j+q <= m; j++)
{
unseen.erase(ha[i][j]);
}
}
int ans=t-unseen.size();
printf("Case %d: %d\n",u, ans);
}
return 0;
}