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萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章