bzoj1127 [POI2008]KUP

Description

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

n<2000
1<=k<=10^9
每个价格都是不大于2*10^9的非负整数

Solution

容易想到先把>2k的格子排除
显然那些本来就满足的单个格子可以直接输出
那么剩下的就是< k的格子了，相当于从01矩阵中找一个满足和>=k的、全为1的最大矩形，上单调栈维护单调不减即可

当k<=sum<=2k时，这是一个合法答案
当sum>2k时，由于每个格子都< k，我们必定可以删掉一行或者一列得到合法答案，即答案一定是找到最大矩形的子矩阵

Code

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef long long LL;
const int N=2005;

LL rc[N][N],a[N];
LL tot[N][N];
LL n,m;
int stack[N],top;

LL cal(int a,int b,int c,int d) {
    return tot[c][d]-tot[a-1][d]-tot[c][b-1]+tot[a-1][b-1];
}

bool output(int a,int b,int c,int d) {
    int tmp=cal(a,b,c,d);
    if (tmp<m) return false;
    if (tmp>=m&&tmp<=m*2) {
        printf("%d %d %d %d\n", b,a,d,c);
        return true;
    }
    if (d-b>c-a) {
        if (output(a+1,b,c,d)) return true;
        if (output(a,b+1,c,d)) return true;
        if (output(a,b,c-1,d)) return true;
        if (output(a,b,c,d-1)) return true;
    } else {
        if (output(a,b,c-1,d)) return true;
        if (output(a,b,c,d-1)) return true;
        if (output(a+1,b,c,d)) return true;
        if (output(a,b+1,c,d)) return true;
    }
    return false;
}

int main(void) {
    scanf("%lld%lld",&m,&n);
    rep(i,1,n) {
        rep(j,1,n) {
            scanf("%d",&rc[i][j]);
            if (rc[i][j]>=m&&rc[i][j]<=m*2) {
                printf("%d %d %d %d\n", j,i,j,i);
                return 0;
            }
            tot[i][j]=tot[i-1][j]+tot[i][j-1]+rc[i][j]-tot[i-1][j-1];
        }
    }
    rep(j,1,n) {
        rep(i,1,n) {
            if (rc[j][i]<m) a[i]++;
            else a[i]=0;
        }
        top=0;
        rep(i,1,n) {
            while (top>0&&a[stack[top]]>=a[i]) {
                if (output(j-a[stack[top]]+1,stack[top-1]+1,j,i-1)) {
                    // solve(j-a[stack[top]]+1,stack[top-1]+1,j,i-1);
                    return 0;
                }
                top--;
            }
            stack[++top]=i;
        }
        while (top>0) {
            if (output(j-a[stack[top]]+1,stack[top-1]+1,j,n)) {
                // solve(j-a[stack[top]]+1,stack[top-1]+1,j,n);
                return 0;
            }
            top--;
        }
    }
    puts("NIE");
    return 0;
}