【整体二分】bzoj2738矩阵乘法

然而传送门并不是去bzoj的ouo

求子矩阵第k小，无修改。

首先将矩阵按照数字大小排序。

我们对所有询问二分答案，将答案以下的数插入树状数组。然后查询。左右分开。

小心变量名冲突。

#include<bits/stdc++.h>
using namespace std;
#define in read()
int in{
	int cnt=0,f=1;char ch=0;
	while(!isdigit(ch)){
		ch=getchar();if(ch=='-')f=-1;
	}
	while(isdigit(ch)){
		cnt=cnt*10+ch-48;
		ch=getchar();
	}return cnt*f;
}
int que[250003],quel[250003],quer[250003];
int n,qq,cnt,ans[250003],t[503][503],cur[250003];
int lowbit(int a){
	return a&(-a);
}
void add(int x,int y,int key){
	int yy=y;
	while(x<=n){
		while(yy<=n){
			t[x][yy]+=key;
			yy+=lowbit(yy);
		}yy=y;x+=lowbit(x);
	}
}
int query(int x,int y){
	int sum=0,yy=y;
	while(x){
		while(yy){
			sum+=t[x][yy];
			yy-=lowbit(yy);
		}x-=lowbit(x);yy=y;
	}return sum;
}
struct node{
	int x,y,key;
}m[250003];
struct bili{
	int x1,y1,x2,y2,k;
}q[250003];
bool cmpp(node a,node b){
	return a.key<b.key;
}
void solve(int l,int r,int ql,int qr){
	//cout<<l<<" "<<r<<" "<<ql<<" "<<qr<<endl;
	if(ql>qr)return;
	if(l==r){
		for(int i=ql;i<=qr;i++)ans[que[i]]=m[l].key;return;
	}
	int mid=(l+r)>>1;
	for(int i=l;i<=mid;i++){
		add(m[i].x,m[i].y,1);
	}
	int cntl=0,cntr=0;
	for(int i=ql;i<=qr;i++){
		int u=que[i],gu=cur[u]+query(q[u].x2,q[u].y2)-query(q[u].x2,q[u].y1-1)-query(q[u].x1-1,q[u].y2)+query(q[u].x1-1,q[u].y1-1);
		if(gu>=q[u].k)quel[++cntl]=u;else quer[++cntr]=u,cur[u]=gu;
	}
	int qcnt=ql-1;
	for(int i=1;i<=cntl;i++)que[++qcnt]=quel[i];
	for(int i=1;i<=cntr;i++)que[++qcnt]=quer[i];
	for(int i=l;i<=mid;i++)add(m[i].x,m[i].y,-1);
	solve(l,mid,ql,ql+cntl-1);solve(mid+1,r,ql+cntl,qr);
}
int main(){
	n=in;qq=in;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			m[++cnt].x=i;m[cnt].y=j;m[cnt].key=in;
		}
	}
	sort(m+1,m+cnt+1,cmpp);
	//for(int i=1;i<=cnt;i++)cout<<m[i].x<<" "<<m[i].y<<endl;
	int z,x,c,v,b;
	for(int i=1;i<=qq;i++)z=in,x=in,c=in,v=in,b=in,q[i]=(bili){z,x,c,v,b};
	for(int i=1;i<=qq;i++)que[i]=i;
	solve(1,cnt,1,qq);
	for(int i=1;i<=qq;i++)printf("%d\n",ans[i]);
	return 0;
}
/*
4 3
2 5 1 3
2 6 4 7
4 3 5 2
4 3 6 4
1 1 4 4 8
1 3 2 4 1
1 2 3 4 2
*/