整体二分+二维树状数组--bzoj2738: 矩阵乘法

传送门

快要被这道题逼疯···

先说正解：

整体二分，与之前不同的是要用二维数据结构，这里二维树状数组就很简单

这里有个讲树状数组的blog不错的http://www.cnblogs.com/RabbitHu/p/BIT.html

套路都是一样的，将询问的答案二分，然后把l~mid之间的数都在二维树状数组上+1，计算询问cnt与k比，将询问分成两部分，然后将树状数组还原，继续二分下去

但是这里每次修改时如果都n^2枚举复杂度会爆炸，因为没有修改，我们可以在最前面把这个矩阵的每个点标号用结构体存下横纵坐标，按数值排序，这样每次从l~mid循环一遍就可以了，最后答案赋成原先这个位置的数值

 

---

再说说我的奇葩tle写法：

因为当时上课的时候yousiki说，如果不会二维数据结构（比如我）可以用前缀和思想，也就是把一个询问拆成两个，就是x2到y轴和x1到y轴之间的差，将询问按照左边排序，这样可以消掉一维，转换成一维的整体二分，然而这样做时间复杂度有点炸···而且很难操作，于是我在尝试了一个多小时无果后只好去学二维树状数组···（原来这么简单）

 

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define N 505
#define M 60005
using namespace std;
int n,m,a[N][N],f[N][N],now1,now2,ans[M],pre[M];

inline int rd(){
	int x=0,f=1;char c=' ';
	while(c<'0' || c>'9') {if(c=='-')f=-1;c=getchar();}
	while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
	return x*f;
}

struct Query{
	int x1,x2,y1,y2,cnt,k,num;
}q[M],q1[M],q2[M];

struct qwq{
	int x,y;
}quq[N*N];
inline bool cmp(qwq x,qwq y) {return a[x.x][x.y]<a[y.x][y.y];}

inline void add(int x,int y,int c){
	while(x<=n){
		for(int i=y;i<=n;i+=i&-i)
			f[x][i]+=c;
		x+=x&-x;
	} return;
}

inline int query(int x,int y){
	int sum=0;
	while(x){
		for(int i=y;i;i-=i&-i)
			sum+=f[x][i];
		x-=x&-x;
	} return sum;
}

inline int range_query(int x1,int y1,int x2,int y2){
	return query(x2,y2)-query(x1-1,y2)-query(x2,y1-1)+query(x1-1,y1-1);
}

inline void calc(int l,int r,int vl,int vr){
	now1=now2=0;
	for(register int i=vl;i<=vr;i++)
		add(quq[i].x,quq[i].y,1);
	for(register int i=l;i<=r;i++){
		q[i].cnt=range_query(q[i].x1,q[i].y1,q[i].x2,q[i].y2);
		if(q[i].cnt>=q[i].k) q1[++now1]=q[i];
		else q[i].k-=q[i].cnt,q2[++now2]=q[i];
	}
	for(register int i=vl;i<=vr;i++)
		add(quq[i].x,quq[i].y,-1);
	int now=l;
	for(register int i=1;i<=now1;i++) q[now++]=q1[i];
	for(register int i=1;i<=now2;i++) q[now++]=q2[i];
	return;
}

inline void solve(int l,int r,int vl,int vr){
	if(vl==vr){
		for(register int i=l;i<=r;i++)
			ans[q[i].num]=a[quq[vl].x][quq[vl].y];
		return;
	}
	int mid=(vl+vr)>>1;
	calc(l,r,vl,mid);
	int tmp=now1;
	if(tmp) solve(l,l+tmp-1,vl,mid);
	if(l+tmp<=r) solve(l+tmp,r,mid+1,vr);
	return;
}

int main(){
	n=rd(); m=rd();
	for(register int i=1;i<=n;i++)
		for(register int j=1;j<=n;j++)
			a[i][j]=rd(),quq[(i-1)*n+j]=(qwq){i,j};
	sort(quq+1,quq+n*n+1,cmp);
	for(register int i=1;i<=m;i++){
		q[i].x1=rd(),q[i].y1=rd(),q[i].x2=rd(),q[i].y2=rd(),q[i].k=rd();
		q[i].num=i;
	}
	solve(1,m,1,n*n);
	for(register int i=1;i<=m;i++)
		printf("%d\n",ans[i]);
	return 0;
}