背景:
原来用
R
M
Q
RMQ
RMQ做的,慢好多。
题目传送门:
https://www.luogu.org/problemnew/show/P2216
题意:
有一个
n
∗
m
n*m
n∗m的矩阵,现在要选出一个
k
∗
k
k*k
k∗k的矩阵,使得这个矩阵的最大值与最小值的差最小,求这个差值。
思路:
用单调队列维护
j
j
j到
j
+
k
−
1
j+k-1
j+k−1的最值。
然后再根据得到的最值,用单调队列维护
i
i
i到
i
+
k
−
1
i+k-1
i+k−1的最值。
这样便得到了所有
k
∗
k
k*k
k∗k的矩阵的最值。
最后在求解即可。
细节较多。
代码:
#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);
}