[bzoj1127]KUP

题目大意

给一个n*n的地图，每个格子有一个价格，找一个矩形区域，使其价格总和位于[k,2k]

做法

首先先排除掉单点的情况，即如果存在一个位置权值是[k,2k]直接输出，接下来所有权值都不在[k,2k]内。
这种情况下，如果有>2k的点，矩形和显然不可能在[k,2k]，因此矩形不能包含>2k的点，所以矩形只能包含 <k <script type="math/tex" id="MathJax-Element-382"> 假如找出一个矩形只包含 <k <script type="math/tex" id="MathJax-Element-383">
为什么呢？考虑下列算法。
如果删除这个矩形第一列仍然是>=k的，就删。
否则，因为和>=2k，删去最后一列即可，也能使矩形仍然>=k。
重复删的过程直至矩形和在[k,2k]内。
当然可能删到只有一列，这时从尾部开始一个一个删。
因为每个元素 <k <script type="math/tex" id="MathJax-Element-384"> <k <script type="math/tex" id="MathJax-Element-385"> 于是我们用单调栈之类的找出每个极大子矩形检验即可。
详见代码。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int maxn=2000+10;
int a[maxn][maxn],up[maxn][maxn],left[maxn],right[maxn];
ll sum[maxn][maxn];
int i,j,k,l,t,n,m,top;
bool czy;
ll getsum(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 calc(int x1,int y1,int x2,int y2){
    if (getsum(x1,y1,x2,y2)<k) return;
    while (getsum(x1,y1,x2,y2)>2*k){
        if (x1==x2) y2--;
        else if (getsum(x1+1,y1,x2,y2)>=k) x1++;
        else x2--;
    }
    printf("%d %d %d %d\n",y1,x1,y2,x2);
    czy=1;
}
void work(int x){
    int i;
    fo(i,1,n){
        j=i-1;
        while (j&&up[x][j]>=up[x][i]) j=left[j]-1;
        left[i]=j+1;
    }
    fd(i,n,1){
        j=i+1;
        while (j<=n&&up[x][j]>up[x][i]) j=right[j]+1;
        right[i]=j-1;
    }
    fo(i,1,n){
        if (i==519){
            t=t;
        }
        if (up[x][i]) calc(x-up[x][i]+1,left[i],x,right[i]);
        if (czy) return;
        //printf("%d\n",i);
    }
}
int main(){
    //freopen("kup5.in","r",stdin);
    scanf("%d%d",&k,&n);
    fo(i,1,n)
        fo(j,1,n){
            scanf("%d",&a[i][j]);
            if (k<=a[i][j]&&a[i][j]<=2*k){
                printf("%d %d %d %d\n",j,i,j,i);
                return 0;
            }
        }
    fo(i,1,n)
        fo(j,1,n)
            sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+(ll)a[i][j];
    fo(i,1,n)
        fo(j,1,n)
            if (a[i][j]>=k) up[i][j]=0;else up[i][j]=up[i-1][j]+1;
    fo(i,1,n){
        work(i);
        if (czy) break;
    }
    if (!czy) printf("NIE\n");
}