bzoj 2738: 矩陣乘法【整體二分+樹狀數組】
阿新 • • 發佈:2019-04-22
ons amp 主席樹 二分 %d r+ 乘法 根據 ostream
腦子一抽開始寫主席樹,敲了一會發現不對……
整體二分,用二維樹狀數組維護值為當前區間的格子個數,然後根據k的大小和當前詢問的子矩陣裏的值和k的大小關系來決定這個詢問放在哪一部分向下遞歸
#include<iostream> #include<cstdio> #include<algorithm> using namespace std; const int N=300005; int n,q,t[505][505],rl[N],ans[N],tot; struct ques { int x,y,xx,yy,k,id; }f[N],g[N]; struct qwe { int i,j,v; }a[N]; bool cmp(const qwe &a,const qwe &b) { return a.v<b.v; } int read() { int r=0,f=1; char p=getchar(); while(p>'9'||p<'0') { if(p=='-') f=-1; p=getchar(); } while(p>='0'&&p<='9') { r=r*10+p-48; p=getchar(); } return r*f; } void update(int x,int y,int v) { for(int i=x;i<=n;i+=(i&(-i))) for(int j=y;j<=n;j+=(j&(-j))) t[i][j]+=v; } int ques(int x,int y) { int r=0; for(int i=x;i>=1;i-=(i&(-i))) for(int j=y;j>=1;j-=(j&(-j))) r+=t[i][j]; return r; } void wk(int ll,int rr,int l,int r) { if(a[ll].v==a[rr].v) { for(int i=l;i<=r;i++) ans[f[i].id]=a[ll].v; return; } int mid=(ll+rr)>>1,c1=l,c2=r; for(int i=ll;i<=mid;i++) update(a[i].i,a[i].j,1); for(int i=l;i<=r;i++) { int nw=ques(f[i].xx,f[i].yy)-ques(f[i].x-1,f[i].yy)-ques(f[i].xx,f[i].y-1)+ques(f[i].x-1,f[i].y-1); if(f[i].k<=nw) g[c1++]=f[i]; else { f[i].k-=nw; g[c2--]=f[i]; } } for(int i=l;i<=r;i++) f[i]=g[i]; for(int i=ll;i<=mid;i++) update(a[i].i,a[i].j,-1); wk(ll,mid,l,c2); wk(mid+1,rr,c1,r); } int main() { n=read(),q=read(); for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) a[++tot]=(qwe){i,j,read()}; sort(a+1,a+1+tot,cmp); for(int i=1;i<=q;i++) f[i].x=read(),f[i].y=read(),f[i].xx=read(),f[i].yy=read(),f[i].k=read(),f[i].id=i; wk(1,tot,1,q); for(int i=1;i<=q;i++) printf("%d\n",ans[i]); return 0; }
bzoj 2738: 矩陣乘法【整體二分+樹狀數組】