【题目链接】 http://www.lydsy.com/JudgeOnline/problem.php?id=2738
【题目大意】
给出一个方格图,询问要求求出矩阵内第k小的元素
【题解】
我们对答案的大小进行整体二分,用二维树状数组维护二维区间和,
将超过数量的分治到左区间,不满足的分治到右区间即可。
【代码】
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=250010,M=510;
namespace Td_BIT{
int c[M][M];
void Initialize(){memset(c,0,sizeof(c));}
void add(int x,int y,int val){
for(int i=x;i<M;i+=i&-i)
for(int j=y;j<M;j+=j&-j)c[i][j]+=val;
}
int query(int x,int y){
int res=0;
for(int i=x;i;i-=i&-i)
for(int j=y;j;j-=j&-j)res+=c[i][j];
return res;
}
}
struct Q{int x1,x2,y1,y2,k;}q[N];
struct data{int x,y,val;}a[N];
bool operator<(data a,data b){return a.val<b.val;}
int cnt;
int query(int k){
using namespace Td_BIT;
int x1=q[k].x1,y1=q[k].y1,x2=q[k].x2,y2=q[k].y2;
return query(x2,y2)+query(x1-1,y1-1)-query(x1-1,y2)-query(x2,y1-1);
}
int mark[N],ans[N],id[N],tmp[N],T=0;
void solve(int l,int r,int L,int R){
if(l>r||L==R)return;
int mid=(L+R)>>1;
while(a[T+1].val<=mid&&T<cnt){Td_BIT::add(a[T+1].x,a[T+1].y,1);T++;}
while(a[T].val>mid){Td_BIT::add(a[T].x,a[T].y,-1);T--;}
int cnt=0;
for(int i=l;i<=r;i++){
if(query(id[i])>q[id[i]].k-1){
mark[i]=1;
ans[id[i]]=mid;
cnt++;
}else mark[i]=0;
}int l1=l,l2=l+cnt;
for(int i=l;i<=r;i++){
if(mark[i])tmp[l1++]=id[i];
else tmp[l2++]=id[i];
}for(int i=l;i<=r;i++)id[i]=tmp[i];
solve(l,l1-1,L,mid);solve(l1,l2-1,mid+1,R);
}
int n,m;
int main(){
scanf("%d%d",&n,&m);
int mx=cnt=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
a[++cnt].x=i; a[cnt].y=j;
scanf("%d",&a[cnt].val);
mx=max(a[cnt].val,mx);
}
}sort(a+1,a+cnt+1);
for(int i=1;i<=m;i++)scanf("%d%d%d%d%d",&q[i].x1,&q[i].y1,&q[i].x2,&q[i].y2,&q[i].k);
for(int i=1;i<=m;i++)id[i]=i;
solve(1,m,0,mx+1);
for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
return 0;
}