BZOJ2738: 矩阵乘法

n<=500,n*n的矩阵，m<=60000个询问每次问子矩阵第K小。

方法一：第K小什么的是可以二分的，那就整体二分套二维树状数组吧！

这里有一个细节优化，本来整体二分是logMax的嘛，可以先把数字排个序，就变成logn了。这是没优化的后果（话说3个log确实极卡）

方法二：分块，n*n个数字从小到大排，然后分n次加入，每次加入n个算贡献，在询问里把这些数字的贡献减掉，如果某个询问的K<=0了，说明他的答案在这几个数里面，暴力查一下即可。n^3+nq也极卡。。

#include<string.h>
#include<stdlib.h>
#include<stdio.h>
//#include<assert.h>
#include<algorithm>
//#include<iostream>
using namespace std;

int n,m;
#define maxn 350011
struct Ope
{
    int x,y,z,w,u,id,type;
    //type=1: xiugai dian x,y wei z
    //type=0: chaxun (x,y) dao (z,w) di u da bianhao id
}q[maxn],al[maxn],ar[maxn]; int len=0;
int ans[maxn];

struct BIT
{
    int a[511][511];
    void add(int x,int y,int v)
    {
        for (;x<=n;x+=x&-x)
            for (int i=y;i<=n;i+=i&-i)
                a[x][i]+=v;
    }
    int query(int x,int y)
    {
        int ans=0;
        for (;x;x-=x&-x)
            for (int i=y;i;i-=i&-i)
                ans+=a[x][i];
        return ans;
    }
}t;

void solve(int L,int R,int ql,int qr)
{
    if (L>R || ql>qr) return;
    if (L==R)
    {
        for (int i=ql;i<=qr;i++) if (q[i].type==0) ans[q[i].id]=L;
        return;
    }
    int mid=(L+R)>>1,lal=0,lar=0;
    for (int i=ql;i<=qr;i++)
    {
        if (q[i].type)
        {
            if (q[i].z<=mid) t.add(q[i].x,q[i].y,1),al[++lal]=q[i];
            else ar[++lar]=q[i];
        }
        else
        {
            int tmp=t.query(q[i].z,q[i].w)-t.query(q[i].z,q[i].y-1)
            -t.query(q[i].x-1,q[i].w)+t.query(q[i].x-1,q[i].y-1);
            if (tmp>=q[i].u) al[++lal]=q[i];
            else q[i].u-=tmp,ar[++lar]=q[i];
        }
    }
    for (int i=1;i<=lal;i++) if (al[i].type) t.add(al[i].x,al[i].y,-1);
    for (int i=1,j=ql;i<=lal;i++,j++) q[j]=al[i];
    for (int i=1,j=ql+lal;i<=lar;i++,j++) q[j]=ar[i];
    solve(L,mid,ql,ql+lal-1);
    solve(mid+1,R,ql+lal,qr);
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        {
            len++; scanf("%d",&q[len].z);
            q[len].x=i; q[len].y=j; q[len].type=1;
        }
    for (int i=1;i<=m;i++) len++,scanf("%d%d%d%d%d",&q[len].x,&q[len].y,&q[len].z,&q[len].w,&q[len].u),
    q[len].id=i,q[len].type=0;
    solve(0,0x3f3f3f3f,1,len);
    for (int i=1;i<=m;i++) printf("%d
",ans[i]);
    return 0;
}