[TK] 理想的正方形
题目描述
有一个整数组成的矩阵,现请你从中找出一个指定边长的正方形区域,使得该区域所有数中的最大值和最小值的差最小。
题目分析
其实这道题和滑动窗口很像,而滑动窗口使用优先队列解决.
我们都知道优先队列可以维护区间最值. 但是优先队列维护的区间不是二维的,而是一维的.
因此,这道题我们的思考方向,一是把优先队列拓展到二维,二是用一维优先队列解决问题.
思考一下怎么用一维优先队列解决问题,我们可以发现,一个区间内的最大值等于各行最大值的最大值,最小值等于各行最小值的最小值. 这启发我们采用先一维再一维的方式解决问题.
因此,我们需要做的是,跑出每行滑动窗口的最值,再跑出相同列滑动窗口最值的最值,在所有最值中选择最优解
代码实现
注: 使用了 封装滑动窗口
MMMW m[1001];
MMMW om[2][1001];
int a,b,n,ans=0x7fffffff;
int main(){
cin>>a>>b>>n;
for(int i=1;i<=a;++i){
for(int j=1;j<=b;++j){
cin>>m[i].a[j];
}
for(int j=1;j<=n-1;++j){
m[i].push(j);
}
for(int j=n;j<=b;++j){
m[i].out(j-n);
m[i].push(j);
om[0][j].a[i]=m[i].max();
om[1][j].a[i]=m[i].min();
}
}
for(int i=n;i<=b;++i){
for(int j=1;j<=n-1;++j){
om[0][i].push(j);
om[1][i].push(j);
}
for(int j=n;j<=a;++j){
om[0][i].out(j-n);
om[0][i].push(j);
om[1][i].out(j-n);
om[1][i].push(j);
ans=min(ans,om[0][i].max()-om[1][i].min());
}
}
cout<<ans;
}