BZOJ 1047 [HAOI2007]理想的正方形 - 单调队列

拆成二维搞一搞就好了撒,还是蛮好想的。

先对于row来一波滑动窗口,记录每个节点在n范围内的最值,然后colomn再搞一遍,求个最值一减ans就出来了(注意colomn搞的时候,>=n才记为有效值,反例是左上角的(1,1)相减答案肯定为0)

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<algorithm>

using namespace std;

const int maxn=1005;

int a,b,n;
long long q[maxn],rmax[maxn][maxn],rmin[maxn][maxn],cmax[maxn],cmin[maxn],s[maxn][maxn];
long long ans=0x7fffffff;

void row(int x)
{
    int head=1,rear=0;
    for(int i=1;i<=b;i++)
    {
        while(head<=rear&&q[head]<=i-n)head++;
        while(head<=rear&&s[x][q[rear]]<s[x][i])rear--;
        q[++rear]=i;
        rmax[x][i]=s[x][q[head]];
    }
    head=1,rear=0;
    for(int i=1;i<=b;i++)
    {
        while(head<=rear&&q[head]<=i-n)head++;
        while(head<=rear&&s[x][q[rear]]>s[x][i])rear--;
        q[++rear]=i;
        rmin[x][i]=s[x][q[head]];
    }
}
void colomn(int x)
{
    int head=1,rear=0;
    for(int i=1;i<=a;i++)
    {
        while(head<=rear&&q[head]<=i-n)head++;
        while(head<=rear&&rmax[q[rear]][x]<rmax[i][x])rear--;
        q[++rear]=i;
        cmax[i]=rmax[q[head]][x];
    }
    head=1,rear=0;
    for(int i=1;i<=a;i++)
    {
        while(head<=rear&&q[head]<=i-n)head++;
        while(head<=rear&&rmin[q[rear]][x]>rmin[i][x])rear--;
        q[++rear]=i;
        cmin[i]=rmin[q[head]][x];
        if(i>=n)ans=min(ans,cmax[i]-cmin[i]);
    }
}
int main()
{
    scanf("%d%d%d",&a,&b,&n);
    for(int i=1;i<=a;i++)
        for(int j=1;j<=b;j++)
            scanf("%lld",&s[i][j]);
    for(int i=1;i<=a;i++)row(i);
    for(int i=n;i<=b;i++)colomn(i);
    printf("%lld",ans);
}
发布了112 篇原创文章 · 获赞 4 · 访问量 1万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章