题目描述
有一个a乘b的整数组成的矩阵,现请你从中找出一个nn的正方形区域,使得该区域所有数中的最大值和最小值的差最小。
输入输出格式
输入格式:
第一行为3个整数,分别表示a,b,n的值
第二行至第a+1行每行为b个非负整数,表示矩阵中相应位置上的数。每行相邻两数之间用一空格分隔。
输出格式:
仅一个整数,为ab矩阵中所有“n*n正方形区域中的最大整数和最小整数的差值”的最小值。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,k,mp[1005][1005];
int dp1[1005][1005][11],dp2[1005][1005][11];
int ans=0x3f3f3f3f;
void st()
{
for(int k=1;(1<<k)<=min(n,m);k++)
for(int i=1;i+(1<<k)-1<=n;i++)
for(int j=1;j+(1<<k)-1<=m;j++)
{
dp1[i][j][k]=max(max(dp1[i][j][k-1],dp1[i+(1<<(k-1))][j][k-1]),max(dp1[i][j+(1<<(k-1))][k-1],dp1[i+(1<<(k-1))][j+(1<<(k-1))][k-1]));
dp2[i][j][k]=min(min(dp2[i][j][k-1],dp2[i+(1<<(k-1))][j][k-1]),min(dp2[i][j+(1<<(k-1))][k-1],dp2[i+(1<<(k-1))][j+(1<<(k-1))][k-1]));
}
}
int RMQ(int x,int y,int xx,int yy)
{
int l=0;
while((1<<(l+1))<=k) l++;
return max(max(dp1[x][y][l],dp1[xx-(1<<l)+1][yy-(1<<l)+1][l]),max(dp1[xx-(1<<l)+1][y][l],dp1[x][yy-(1<<l)+1][l]))-min(min(dp2[x][y][l],dp2[xx-(1<<l)+1][yy-(1<<l)+1][l]),min(dp2[xx-(1<<l)+1][y][l],dp2[x][yy-(1<<l)+1][l]));
}
int main()
{
cin>>n>>m>>k;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
cin>>mp[i][j];
dp1[i][j][0]=dp2[i][j][0]=mp[i][j];//初始化不能忘
}
st();
for(int i=1;i+k-1<=n;i++)
for(int j=1;j+k-1<=m;j++)
ans=min(ans,RMQ(i,j,i+k-1,j+k-1));
cout<<ans<<endl;
}