码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ_2738_矩阵乘法_整体二分

时间:2018-04-28 21:06:38      阅读:153      评论:0      收藏:0      [点我收藏+]

标签:++   div   tput   define   content   sample   查询   ons   return   

BZOJ_2738_矩阵乘法_整体二分

Description

  给你一个N*N的矩阵,不用算矩阵乘法,但是每次询问一个子矩形的第K小数。

Input

  第一行两个数N,Q,表示矩阵大小和询问组数;
  接下来N行N列一共N*N个数,表示这个矩阵;
  再接下来Q行每行5个数描述一个询问:x1,y1,x2,y2,k表示找到以(x1,y1)为左上角、以(x2,y2)为右下角的子矩形中的第K小数。

Output

  对于每组询问输出第K小的数。

Sample Input

2 2
2 1
3 4
1 2 1 2 1
1 1 2 2 3

Sample Output

1
3

HINT

  矩阵中数字是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]);
}

 

BZOJ_2738_矩阵乘法_整体二分

标签:++   div   tput   define   content   sample   查询   ons   return   

原文地址:https://www.cnblogs.com/suika/p/8967222.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!