luogu P2216 [HAOI2007]理想的正方形

背景:

原来用 R M Q RMQ RMQ做的,慢好多。

题目传送门:

https://www.luogu.org/problemnew/show/P2216

题意:

有一个 n ∗ m n*m nm的矩阵,现在要选出一个 k ∗ k k*k kk的矩阵,使得这个矩阵的最大值与最小值的差最小,求这个差值。

思路:

用单调队列维护 j j j j + k − 1 j+k-1 j+k1的最值。
然后再根据得到的最值,用单调队列维护 i i i i + k − 1 i+k-1 i+k1的最值。
这样便得到了所有 k ∗ k k*k kk的矩阵的最值。
最后在求解即可。
细节较多。

代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
	int n,m,k,ans=2147483647;
	int a[1010][1010];
	int ma[1010][1010],mi[1010][1010],ans1[1010][1010],ans2[1010][1010];
	int que1[1010],que2[1010];
void work1()
{
	for(int i=1;i<=n;i++)
	{
		int head1=1,tail1=0,head2=1,tail2=0;
		for(int j=1;j<=m;j++)
		{
			while(head1<=tail1&&a[i][que1[tail1]]<a[i][j]) tail1--;
			que1[++tail1]=j;
			while(head2<=tail2&&a[i][que2[tail2]]>a[i][j]) tail2--;
			que2[++tail2]=j;
			if(j>=k)
			{
				ma[i][j-k+1]=a[i][que1[head1]];
				mi[i][j-k+1]=a[i][que2[head2]];
				while(head1<=tail1&&j-que1[head1]+1>=k) head1++;
				while(head2<=tail2&&j-que2[head2]+1>=k) head2++;
			}
		}
	}
}
void work2()
{
	for(int j=1;j<=m-k+1;j++)
	{
		int head1=1,tail1=0,head2=1,tail2=0;
		for(int i=1;i<=n;i++)
		{
			while(head1<=tail1&&ma[que1[tail1]][j]<ma[i][j]) tail1--;
			que1[++tail1]=i;
			while(head2<=tail2&&mi[que2[tail2]][j]>mi[i][j]) tail2--;
			que2[++tail2]=i;
			if(i>=k)
			{
				ans1[i-k+1][j]=ma[que1[head1]][j];
				ans2[i-k+1][j]=mi[que2[head2]][j];
				while(head1<=tail1&&i-que1[head1]+1>=k) head1++;
				while(head2<=tail2&&i-que2[head2]+1>=k) head2++;
			}
		}
	}
}
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",&a[i][j]);
	work1();
	work2();
	for(int i=1;i<=n-k+1;i++)
		for(int j=1;j<=m-k+1;j++)
			ans=min(ans,ans1[i][j]-ans2[i][j]);
	printf("%d",ans);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值