理想的正方形 单调队列优化DP

题目:

有一个 a×b 的整数组成的矩阵,现请你从中找出一个 n×n 的正方形区域,使得该区域所有数中的最大值和最小值的差最小。

思路:

对每一行使用单调队列求区间内的最小值和最大值并存储起来,然后对列方向分别求最小值和最大值。最终把每个 n × n n×n n×n的方格的最小值和最大值存储在右下角的位置。然后遍历取答案的最小即可。

代码:

#include<bits/stdc++.h>
using namespace std;
const int N = 1010;

int w[N][N];
int mn[N][N],mx[N][N];
int q[N];
int n,m,k;

void get_min(int a[],int b[],int cnt)
{
	int hh=0,tt=-1;
	for(int i=1;i<=cnt;i++)
	{
		if(hh<=tt && q[hh] <= i-k) ++hh;
		while(hh<=tt && a[i] <= a[q[tt]]) --tt;
		q[++tt] = i;
		b[i] = a[q[hh]];
	}
}
void get_max(int a[],int b[],int cnt)
{
	int hh=0,tt=-1;
	for(int i=1;i<=cnt;i++)
	{
		if(hh<=tt && q[hh] <= i-k) ++hh;
		while(hh<=tt && a[i] >= a[q[tt]]) --tt;
		q[++tt] = i;
		b[i] = a[q[hh]];
	}
}
int main()
{
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			scanf("%d",&w[i][j]);
	//对每一行求
	for(int i=1;i<=n;i++)
	{
		get_max(w[i],mx[i],m);
		get_min(w[i],mn[i],m);
	}
	//在列的方向求最小值和最大值
	int res = 1e9;
	int a[N],b[N],c[N];
	for(int j=k;j<=m;j++)
	{
		for(int i=1;i<=n;i++) a[i] = mn[i][j];
		get_min(a,b,n);
		for(int i=1;i<=n;i++) a[i] = mx[i][j];
		get_max(a,c,n);
		for(int i=k;i<=n;i++)
			res = min(res,c[i]-b[i]);
	}
	printf("%d\n",res);
	return 0;
}
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

行码棋

码字好辛苦,总结好吃力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值