1567: [JSOI2008]Blue Mary的战役地图
Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 1199 Solved: 702
[Submit][Status][Discuss]
Description
Blue Mary最近迷上了玩Starcraft(星际争霸) 的RPG游戏。她正在设法寻找更多的战役地图以进一步提高自己的水平。 由于Blue Mary的技术已经达到了一定的高度,因此,对于用同一种打法能够通过的战役地图,她只需要玩一张,她就能了解这一类战役的打法,然后她就没有兴趣再玩儿这一类地图了。而网上流传的地图有很多都是属于同一种打法,因此Blue Mary需要你写一个程序,来帮助她判断哪些地图是属于同一类的。 具体来说,Blue Mary已经将战役地图编码为n*n的矩阵,矩阵的每个格子里面是一个32位(有符号)正整数。对于两个矩阵,他们的相似程度定义为他们的最大公共正方形矩阵的边长。两个矩阵的相似程度越大,这两张战役地图就越有可能是属于同一类的。
Input
第一行包含一个正整数n。 以下n行,每行包含n个正整数,表示第一张战役地图的代表矩阵。 再以下n行,每行包含n个正整数,表示第二张战役地图的代表矩阵。
Output
仅包含一行。这一行仅有一个正整数,表示这两个矩阵的相似程度。
Sample Input
3
1 2 3
4 5 6
7 8 9
5 6 7
8 9 1
2 3 4
Sample Output
2
HINT
样例解释:
子矩阵:
5 6
8 9
为两个地图的最大公共矩阵
约定:
n<=50
二维哈希,之后枚举边长找是否有相同
代码:
#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 Hash1[maxn][maxn], Hash2[maxn][maxn];
int a[maxn][maxn], b[maxn][maxn];
int data[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;
}
}
int main() {
int n;
scanf("%d", &n);
init();
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++)
scanf("%d", &a[i][j]);
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++)
scanf("%d", &b[i][j]);
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
Hash1[i][j] += Hash1[i][j - 1] * seed1 + a[i][j]; //列哈希 前缀和
Hash2[i][j] += Hash2[i][j - 1] * seed1 + b[i][j];
}
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
Hash1[i][j] += Hash1[i - 1][j] * seed2; //行哈希 前缀和
Hash2[i][j] += Hash2[i - 1][j] * seed2;
}
}
for (int k = n; k > 0; k--) {
memset(data, 0, sizeof(data));
int num = 0;
for (int i = k; i <= n; i++) {
for (int j = k; j <= n; j++) {
int v = (int) (Hash1[i][j] - Hash1[i - k][j] * f2[k] - Hash1[i][j - k] * f1[k] +
Hash1[i - k][j - k] * f1[k] * f2[k]) % mod;
data[num++] = v;
}
}
sort(data, data + num);
for (int i = k; i <= n; i++) {
for (int j = k; j <= n; j++) {
int v = (int) (Hash2[i][j] - Hash2[i - k][j] * f2[k] - Hash2[i][j - k] * f1[k] +
Hash2[i - k][j - k] * f1[k] * f2[k]) % mod;
if (data[lower_bound(data, data + num, v) - data] == v) {
printf("%d\n", k);
return 0;
}
}
}
}
printf("0\n");
return 0;
}
上面的代码直接循环找,还有一个省时间的就是二分。写题是一看数据可以暴力就再没想别的
哈希+二分代码:
#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 Hash1[maxn][maxn], Hash2[maxn][maxn];
int a[maxn][maxn], b[maxn][maxn];
ll data[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;
}
}
bool check(int k, int n) {
int num = 0;
for (int i = k; i <= n; i++) {
for (int j = k; j <= n; j++) {
ll v = Hash1[i][j] - Hash1[i - k][j] * f2[k] - Hash1[i][j - k] * f1[k] +
Hash1[i - k][j - k] * f1[k] * f2[k];
data[num++] = v;
}
}
sort(data, data + num);
for (int i = k; i <= n; i++) {
for (int j = k; j <= n; j++) {
ll v = Hash2[i][j] - Hash2[i - k][j] * f2[k] - Hash2[i][j - k] * f1[k] +
Hash2[i - k][j - k] * f1[k] * f2[k];
if (data[lower_bound(data, data + num, v) - data] == v) {
return true;
}
}
}
return false;
}
int main() {
int n;
scanf("%d", &n);
init();
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++)
scanf("%d", &a[i][j]);
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++)
scanf("%d", &b[i][j]);
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
Hash1[i][j] += Hash1[i][j - 1] * seed1 + a[i][j]; //列哈希 前缀和
Hash2[i][j] += Hash2[i][j - 1] * seed1 + b[i][j];
}
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
Hash1[i][j] += Hash1[i - 1][j] * seed2; //行哈希 前缀和
Hash2[i][j] += Hash2[i - 1][j] * seed2;
}
}
int l = 1, r = n, ans = 0;
while (l <= r) {
int mid = (l + r) >> 1;
if (check(mid, n)) ans = mid, l = mid + 1;
else r = mid - 1;
}
printf("%d\n", ans);
return 0;
}