字符串哈希+二分+dp-Rectangle-HDU4495

字符串哈希+二分+dp-Rectangle-HDU4495

题意：

$$\begin{gathered} T组测试用例，每组包括一个n\times m的由小写字母组成的矩阵。 \\ 定义“等腰直角三角形”：直角边平行于矩阵的边，且关于直角角平分线对称，即每一层直角边均相同。 \\ 要计算整个矩阵中，最大的等腰直角三角形的面积(字母数量)。 \end{gathered}$$

样例：

Sample Input:
1 
4 4 
abab 
dacb 
adab 
cabb

Sample Output:`在这里插入代码片`
6

$容易找到一个最大的等腰直角三角形，直角顶点在(2,1)位置。$

数据范围：

$$\begin{gathered} T\in [1,20]，n,m\in [1,500]。 \\ TimeLimit:100000ms,MemoryLimit:102400K。 \end{gathered}$$

题解：

$$\begin{gathered} ①、根据直角的位置来枚举最长的直角边，有四种可能。 \\ 可以只处理一种直角情况，剩下的三种可以通过把矩阵旋转90度来解决。 \\ ②、这里只处理直角顶点在“右下方”的情况，设顶点坐标为(i,j)，mlen[i][j]为该顶点处的最大相等直角边的长度， \\ dp[i][j]为以该点为顶点，所能构成的最大等腰直角三角形的直角边的长度， \\ 有状态转移方程dp[i+1][j+1]=min(dp[i][j]+2,mlen[i+1][j+1])，如下图： \end{gathered}$$
$$\begin{gathered} ③、那么每一趟用ans保存所有dp[i][j]中最大值即所能构成的最大等腰直角三角形的边长， \\ 再将矩形翻转90°做同样的操作，共执行4次。 \end{gathered}$$

$④、最终面积通过数列求和公式:\frac{(ans+1)(ans)}{2}得到。$

具体落实：

$$\begin{gathered} ①、对每个输入样例分行和列求哈希值保存到hr和hc中。 \\ ②、对每个点(i,j)求最大相同直角边长，得到数组mlen。这一部分用哈希来判断相等，用二分处理最大长度。 \\ ③、dp求每个点作为直角顶点所能构成的最大等腰直角三角形的边长，用ans保存所有结果中的最大值。 \\ ④、矩阵旋转90°，共执行4次，最后利用公式求最大面积。 \end{gathered}$$

注意：

$①、处理每一列的哈希值时，是当作把矩阵逆时针旋转90°来看待的。哈希的区间如下图：$

$②、旋转矩阵后，行与列的值发生交换。$

---

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<map>
#define ll long long
using namespace std;
const int N=510;
const int base=131;
const int mod=1e9+7;
char s[N][N];
int n,m,T;
int mlen[N][N],dp[N][N],ans;  ///mlen[i][j] ：(i,j)所在位置最长对称直角边的长度
ll hr[N][N],hc[N][N],p[N];

ll get(ll h[],int l,int r)
{
    return (h[r]-h[l-1]*p[r-l+1]%mod+mod)%mod;
}

void get_hash()
{
    for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                hr[i][j]=(hr[i][j-1]*base%mod+s[i][j])%mod;

    for(int i=1;i<=m;i++)
            for(int j=1;j<=n;j++)
                hc[i][j]=(hc[i][j-1]*base%mod+s[j][i])%mod;
}

bool check(int i,int j,int len)
{
    return (get(hr[i],j-len+1,j)==get(hc[j],i-len+1,i)) ? true : false ;
}

void cal()   ///二分预处理mlen数组
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            int l=1,r=min(i,j);
            while(l<r)
            {
                int mid=(l+r+1)>>1;
                if(!check(i,j,mid)) r=mid-1;
                else l=mid;
            }
            mlen[i][j]=l;
        }
}

void solve()
{
    get_hash();
    cal();
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            dp[i][j]=min(dp[i-1][j-1]+2,mlen[i][j]);
            ans=max(ans,dp[i][j]);
        }
}

void Rotate()
{
    int tmp[N][N];
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            tmp[j][n-i+1]=s[i][j];

    swap(n,m);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            s[i][j]=tmp[i][j];
}

int main()
{
    p[0]=1;
    for(int i=1;i<N;i++) p[i]=p[i-1]*base%mod;

    scanf("%d",&T);
    while(T--)
    {
        ans=0;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) scanf("%s",s[i]+1);

        for(int i=0;i<4;i++)
        {
            solve();
            Rotate();
        }

        printf("%d\n",(ans+1)*ans/2);
    }

    return 0;
}