题目大意
给一个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");
}