【BZOJ】2738: 矩阵乘法-整体二分+二维BIT

传送门：bzoj2738

---

题解

如标签。

注意常数问题。如把点排序后区间加(不需要每次$n^2$判断哪些点值$\le mid$），同样$BIT$的还原也是单点-1(而不是$memset$)。

还有就是二分的时候循环变量写在外面注意不要混用(RE/WA)

---

代码

#include<bits/stdc++.h>
using namespace std;
const int N=502,M=6e4+5;

int n,m,vl[250005],tot,tre[N][N];
int v[N][N],ans[M],num;

char cp,OS[100];
inline void rd(int &x)
{
    cp=getchar();x=0;
    for(;!isdigit(cp);cp=getchar());
    for(;isdigit(cp);cp=getchar()) x=(x<<3)+(x<<1)+(cp^48);
}

inline void ot(int x)
{
    int re=0;for(;(!re)||(x);x/=10) OS[++re]='0'+x%10;
    for(OS[0]='\n';~re;--re) putchar(OS[re]);
}

struct qr{
    int xa,ya,xb,yb,K,id;
    inline void ini(){rd(xa);rd(ya);rd(xb);rd(yb);rd(K);}
}q[M],res[M];
struct P{
  int x,y,v;
  bool operator<(const P&ky)const{return v<ky.v;}
}p[250005];

inline void ad(int x,int y,int v)
{
	int i;
	for(;x<=n;x+=(x&(-x)))
	 for(i=y;i<=n;i+=(i&(-i)))
	  tre[x][i]+=v;
}

inline int ask(int x,int y)
{
	int re=0,i;
	for(;x;x-=(x&(-x)))
	 for(i=y;i;i-=(i&(-i)))
	  re+=tre[x][i];
	return re;
}

inline int gt(qr r)
{return ask(r.xb,r.yb)-ask(r.xa-1,r.yb)-ask(r.xb,r.ya-1)+ask(r.xa-1,r.ya-1);}

void sol(int l,int r,int st,int ed,int fr,int to)
{
	if(l==r) {for(;st<=ed;++st) ans[q[st].id]=vl[l];return;}
	int mid=(l+r)>>1,i,j,k,pos,ty;
	for(i=fr;(i<=to) && (p[i].v<=mid);++i) ad(p[i].x,p[i].y,1);pos=i-1;
	for(i=st,j=ed,k=st;k<=ed;++k){
	  ty=gt(q[k]);
	  if(q[k].K<=ty) res[i++]=q[k];
	  else {q[k].K-=ty;res[j--]=q[k];}
	}
	for(k=fr;k<=pos;++k) ad(p[k].x,p[k].y,-1);
	for(k=st;k<i;++k) q[k]=res[k];for(k=j+1;k<=ed;++k) q[k]=res[k];
	if(i>st) sol(l,mid,st,i-1,fr,pos);
	if(j<ed) sol(mid+1,r,j+1,ed,pos+1,to);
}

int main(){
	int i,j;rd(n);rd(m);
	for(i=1;i<=n;++i) for(j=1;j<=n;++j){
		rd(v[i][j]);p[++tot]=(P){i,j,v[i][j]};
	}
	sort(p+1,p+tot+1);vl[0]=-1;
	for(i=1;i<=tot;++i){
		if(p[i].v!=vl[num]) {num++;vl[num]=p[i].v;}
		p[i].v=num;
	}
	for(i=1;i<=m;++i) q[i].ini(),q[i].id=i;
	sol(1,num,1,m,1,tot);
	for(i=1;i<=m;++i) ot(ans[i]);
	return 0;
}