【BZOJ2738】矩阵乘法（整体二分+二位树状数组）

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

　　矩阵中数字是10^9以内的非负整数；
　　20%的数据：N<=100,Q<=1000；
　　40%的数据：N<=300,Q<=10000；
　　60%的数据：N<=400,Q<=30000；
　　100%的数据：N<=500,Q<=60000。

题解：
又是一道整体二分。和Meteors思路有相似之处，二分出来不到k个的放左边，大于的放右边。
代码如下：

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#define ll long long
#define inf 0x7f7f7f7f
#define N 505
#define Q 60005 
#define lb(x) (x&(-x))
using namespace std;
ll read()
{
    ll x=0;
    char c=getchar();
    while(c<'0' || c>'9') c=getchar();
    while(c<='9' && c>='0') {x=x*10+c-'0';c=getchar();}
    return x;
}
int n,m,cnt,T,t[N][N];
int id[Q],ans[Q],tmp[Q];
bool vis[Q];
struct nod
{
    int x,y,val;
    bool operator < (const nod& tmp) const{return val<tmp.val;}
}a[N*N];
struct que
{
    int x,y,x2,y2,k;
}q[Q];
void add(int x,int y,int v)
{
    int tmp=y;
    while(x<=n)
    {
        while(y<=n)
        {
            t[x][y]+=v;
            y+=lb(y);
        }
        x+=lb(x);y=tmp;
    }
}
int query(int x,int y)
{
    int ret=0,tmp=y;
    while(x)
    {
        while(y)
        {
            ret+=t[x][y];
            y-=lb(y);
        }
        x-=lb(x);y=tmp;
    }
    return ret;
}
int query(int k)
{
    int x=q[k].x,y=q[k].y,x2=q[k].x2,y2=q[k].y2;
    return query(x2,y2)-query(x-1,y2)-query(x2,y-1)+query(x-1,y-1);
}
void solve(int l,int r,int L,int R)
{
    if(l>r || L==R) return ;
    int mid=(L+R)>>1;
    while(a[T+1].val<=mid && T<cnt) T++,add(a[T].x,a[T].y,1);
    while(a[T].val>mid) add(a[T].x,a[T].y,-1),T--;
    int cnt=0;
    for(int i=l;i<=r;i++)
    {
        if(query(id[i])>q[id[i]].k-1)
        {
            vis[i]=1;
            ans[id[i]]=mid;
            cnt++;
        }
        else vis[i]=0;
    }
    int l1=l,l2=l+cnt;
    for(int i=l;i<=r;i++)
    {
        if(vis[i]) tmp[l1++]=id[i];
        else tmp[l2++]=id[i];
    }
    for(int i=l;i<=r;i++) id[i]=tmp[i];
    solve(l,l1-1,L,mid);solve(l1,l2-1,mid+1,R);
}
int main()
{
    n=read(),m=read();
    int mx=0;
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++) 
    {
        a[++cnt].x=i,a[cnt].y=j,a[cnt].val=read();
        mx=max(mx,a[cnt].val);
    }
    sort(a+1,a+cnt+1);
    for(int i=1;i<=m;i++) q[i].x=read(),q[i].y=read(),q[i].x2=read(),q[i].y2=read(),q[i].k=read();
    for(int i=1;i<=m;i++) id[i]=i;
    solve(1,m,0,mx+1);
    for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
    return 0;
}