NOIP 考前 队列复习

BZOJ 1127

 
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 2020
using namespace std;
int n,k,a[M][M];
long long sum[M][M];
long long Get_Sum(int x1,int y1,int x2,int y2)
{
    return sum[x2][y2]-sum[x1-1][y2]-sum[x2][y1-1]+sum[x1-1][y1-1];
}
void Output(int x1,int y1,int x2,int y2)
{
    while( Get_Sum(x1,y1,x2,y2)>k*2 )
    {
        if(x1==x2)
            y2--;
        else
        {
            if( Get_Sum(x1+1,y1,x2,y2)>=k )
                x1++;
            else
                x2--;
        }
    }
    printf("%d %d %d %d
",y1,x1,y2,x2);
    exit(0);
}
void Monotonous_Stack(int base,int h[])
{
    static int stack[M],top;
    static int l[M],r[M];
    int i;
 
    for(top=0,i=1;i<=n+1;i++)
    {
        while( top && h[stack[top]]>h[i] )
            r[stack[top--]]=i-1;
        stack[++top]=i;
    }
    for(top=0,i=n;i>=0;i--)
    {
        while( top && h[stack[top]]>h[i] )
            l[stack[top--]]=i+1;
        stack[++top]=i;
    }
    for(i=1;i<=n;i++)
        if(h[i])
        {
            long long temp=Get_Sum(base-h[i]+1,l[i],base,r[i]);
            if( temp>=k )
                Output(base-h[i]+1,l[i],base,r[i]);
        }
}
int main()
{
    int i,j;
    cin>>k>>n;
    for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
        {
            scanf("%d",&a[i][j]);
            if( a[i][j]>=k && a[i][j]<=k*2 )
            {
                printf("%d %d %d %d
",j,i,j,i);
                return 0;
            }
            sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j];
        }
    static int h[M];
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++)
            h[j]=a[i][j]>k*2?0:h[j]+1;
        Monotonous_Stack(i,h);
    }
    puts("NIE");
    return 0;
}