【二分&前缀和】UPC Contest2533 - 2020年夏混合个人训练第七十九场 B--Wave

问题 B: Wave

时间限制: 1 Sec 内存限制: 128 MB

题目描述
波在不同的介质中的传播速度是不一样的。真空中波速都是$3\ast 10^{8}m/s$，而在液体介质中的波速会比真空中的波速小，并且在不同的液体介质中波速不一样。我们将一个液体表面分成$N\ast M$块相同大小的正方形格子，每个格子中只包含一种液体介质。现在我们想知道，从某个波源传出去的波能以不变的波速延伸到多大的以波源为中心的大正方形内。假定所有大正方形的边都是平行于坐标轴的。

输入
第一行包含三个整数$N、M和Q$，分别表示液体表面分成的正方形格子的行数和列数，以及询问的个数。接下来是一个$N\ast M$的矩阵，其中每个元素包含一个小写字母表示相应各自内的介质，不同的字母表示不同的介质。接下来$Q$行每行用两个数$X$和$Y$描述一个询问，表示询问以第$X$行第$Y$列为波源的波能以不变的波速延伸到的最大的以波源为中心的大正方形的边长。注意行的标号从$0$到$N-1$，列的标号从$0$到$M-1$。

输出
输出包括 $Q$ 行，每行包含一个整数，即相应问题的答案。

样例输入 Copy

5 5 3
abbaa
abbaa
aaaaa
aaaaa
aaaaa
1 2
1 4
3 2

样例输出 Copy

1
1
3

提示
对于$100\%$的数据，有$1\le N,M\le 1000$，$1\le Q\le 10000$。

题目大意：
给出矩阵，Q组询问，每次询问给出一对（x,y），找出以（x,y）为中心的最大的正方形，且正方形内的所有元素和（x,y）处的元素是一致的。输出最大正方形的边长。

解题思路：
二分距当前位置的距离，判断该距离是否符合条件，其中，这里这个check需要，特殊处理一下，因为有Q此询问，如果暴力查找的话，时间复杂度O(mnlogn*Q),超过1e8了，过不去。因此，想办法O(1)查找，需要预处理出26个字母矩阵前缀和，也就是说，可以O（1）的时间判断某个连续子矩阵中有多少个相应的字母。之后再和边长的平方作比较。

上代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 1010;
int n,m,q;
char a[N][N];
int s[30][N][N];
bool check(int len,int x,int y)
{
    int k=a[x][y]-'a';
    if(x-len<1||x+len>n||y-len<1||y+len>m) return false;
    int x2 = x + len,y2 = y + len;//右下角 
    int x1 = x - len,y1 = y - len;//左上角
    int tmp = s[k][x2][y2] - s[k][x2][y1 - 1] - s[k][x1 - 1][y2] + s[k][x1 - 1][y1 - 1];
    int b = 2 * len + 1;
    if(tmp == b * b) return true;
    else return false;
}
int main()
{
    cin >> n >> m >> q;
    for(int i = 1;i <= n;i ++) cin >> a[i] + 1;
    for(int k = 0;k < 26;k ++)//对26个字母分别求前缀和  
    {
        for(int i = 1;i <= n;i ++)
        {
            for(int j = 1;j <= m;j ++)
            {
                s[k][i][j]=s[k][i-1][j]+s[k][i][j-1]-s[k][i-1][j-1]+(a[i][j]-'a'==k); 
            }
        }
    } 
    while(q--)
    {
        int x,y;
        cin >> x >> y;
        x++;y++;
        int l = 0,r = max(m,n),mid;
        while(l < r)
        {
            mid = (l + r + 1) >> 1;
            if(check(mid,x,y))
            {
                l = mid;
            }
            else r = mid - 1;
        }
        int ans = 2 * l + 1;
        printf("%d\n",ans);
    }
    return 0;
}