[国家集训队]矩阵乘法

题面

题意

给出一个矩形，每次询问子矩形中第k小的值。

做法

首先考虑二分答案，但是查询矩形中比某个数小的个数的复杂度极高，因此我们可以考虑整体二分。
整体二分：对于所有询问一起二分，处理时将答案小于等于mid的询问放左边，答案大于mid的询问放右边，然后递归分别处理左右两边，计算小于等于mid的数时，可以用二维树状数组动态完成，这样只要每次对此时处理的询问都询问一次二维前缀和来判断答案位置，然后据此将其分组即可。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define P pair<int,int>
#define mp make_pair
#define fi first
#define se second
#define N 510
#define M 60010
#define MN 250010
using namespace std;

int n,m,num[N][N],ans[M],sz[N][N],tt,cnt;
P b[MN];
struct Que
{
    int a,b,c,d,e,id;
}que[M],tmp[M];

inline int get(P u){return num[u.fi][u.se];}
inline bool cmp(P u,P v){return get(u)<get(v);}
inline int lb(int u){return u&(-u);}

inline void add(P w,int t)
{
    int u,v;
    for(u=w.fi;u<=n;u+=lb(u))
    {
        for(v=w.se;v<=n;v+=lb(v))
        {
            sz[u][v]+=t;
        }
    }
}

inline int ask(P w)
{
    int u,v,res=0;
    for(u=w.fi;u;u-=lb(u))
    {
        for(v=w.se;v;v-=lb(v))
        {
            res+=sz[u][v];
        }
    }
    return res;
}

void dfs(int l,int r,int u,int v)
{
    if(u>v) return;
    if(l==r)
    {
        int i,j;
        for(i=u;i<=v;i++)
        {
            ans[que[i].id]=get(b[l]);
        }
        return;
    }
    int i,j,p=u-1,q=v+1,mid=((l+r)>>1);
    for(i=l;i<=mid;i++) add(b[i],1);
    for(i=u;i<=v;i++)
    {
        if(ask(mp(que[i].c,que[i].d))+ask(mp(que[i].a-1,que[i].b-1))-ask(mp(que[i].c,que[i].b-1))-ask(mp(que[i].a-1,que[i].d))>=que[i].e) tmp[++p]=que[i];
        else tmp[--q]=que[i];
    }
    for(i=u;i<=v;i++) que[i]=tmp[i];
    dfs(mid+1,r,q,v);
    for(i=l;i<=mid;i++) add(b[i],-1);
    dfs(l,mid,u,p);
}

int main()
{
    int i,j;
    cin>>n>>m;
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++)
        {
            scanf("%d",&num[i][j]);
            b[++tt]=mp(i,j);
        }
    }
    sort(b+1,b+tt+1,cmp);
    for(i=1;i<=m;i++)
    {
        scanf("%d%d%d%d%d",&que[i].a,&que[i].b,&que[i].c,&que[i].d,&que[i].e);
        que[i].id=i;
    }
    dfs(1,tt,1,m);
    for(i=1;i<=m;i++) printf("%d\n",ans[i]);
}