bzoj 1567: [JSOI2008]Blue Mary的战役地图 二维哈希

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;
}