给你一个N*N的矩阵,不用算矩阵乘法,但是每次询问一个子矩形的第K小数。
标签:++ div tput define content sample 查询 ons return
BZOJ_2738_矩阵乘法_整体二分
矩阵中数字是109以内的非负整数;
20%的数据:N<=100,Q<=1000;
40%的数据:N<=300,Q<=10000;
60%的数据:N<=400,Q<=30000;
100%的数据:N<=500,Q<=60000。
可以离线,把权值排序。
solve(b,e,l,r)表示b到e的询问的答案在l到r范围的权值内。
答案mid,就插入前mid个数,然后查询区间有多少个数,可以用二维树状数组维护。
答案在左边就往左走,在右边就减去左边的贡献往右走。
代码:
#include <stdio.h> #include <string.h> #include <algorithm> using namespace std; #define N 60050 int n,c[550][550],L[N],R[N],mid[N],ans[N]; struct A { int v,x,y; }a[550*550]; struct Q { int d,f,g,h,k,id; }q[N],t[N]; bool cmp(const A &a,const A &b) { return a.v<b.v; } void fix(int x,int y,int v) { int i,j; for(i=x;i<=n;i+=i&(-i)) { for(j=y;j<=n;j+=j&(-j)) { c[i][j]+=v; } } } int inq(int x,int y) { int i,j,re=0; for(i=x;i;i-=i&(-i)) { for(j=y;j;j-=j&(-j)) { re+=c[i][j]; } } return re; } int query(int x,int y,int x2,int y2) { x--; y--; return inq(x2,y2)-inq(x,y2)-inq(x2,y)+inq(x,y); } void solve(int b,int e,int l,int r) { int i; if(b>e) return ; if(l==r) { for(i=b;i<=e;i++) { ans[q[i].id]=a[l].v; } return ; } int mid=(l+r)>>1,lpos=b,rpos=e; for(i=l;i<=mid;i++) { fix(a[i].x,a[i].y,1); } for(i=b;i<=e;i++) { int sizls=query(q[i].d,q[i].f,q[i].g,q[i].h); if(sizls>=q[i].k) t[lpos++]=q[i]; else q[i].k-=sizls,t[rpos--]=q[i]; } for(i=b;i<=e;i++) q[i]=t[i]; for(i=l;i<=mid;i++) { fix(a[i].x,a[i].y,-1); } solve(b,lpos-1,l,mid); solve(rpos+1,e,mid+1,r); } int main() { int m; scanf("%d%d",&n,&m); int i,tot=0,j; for(i=1;i<=n;i++) { for(j=1;j<=n;j++) { scanf("%d",&a[++tot].v); a[tot].x=i; a[tot].y=j; } } sort(a+1,a+n*n+1,cmp); for(i=1;i<=m;i++) { scanf("%d%d%d%d%d",&q[i].d,&q[i].f,&q[i].g,&q[i].h,&q[i].k); q[i].id=i; } solve(1,m,1,n*n); for(i=1;i<=m;i++) printf("%d\n",ans[i]); }
标签:++ div tput define content sample 查询 ons return
原文地址:https://www.cnblogs.com/suika/p/8967222.html