洛谷P1527 [国家集训队]矩阵乘法

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小的数。

---

Solution

题目描述离谱。
不强制在线，意味着可以乱搞。
Q个询问直接离线整体二分，二维树状数组维护答案。

---

Code

#include<bits/stdc++.h>
using namespace std;
struct data{
    int x,y,s;
}a[250010];
struct qry{
    int x1,y1,x2,y2,s,id;
}q[100010],q1[100010],q2[100010];
int n,m,now;
int sum[510][510];
int ans[100010];
bool cmp(data u,data v){
    return u.s<v.s;  
}
int lowbit(int x){
    return x&-x;
}
void add(int x,int y,int s){
    for(int i=x;i<=n;i+=lowbit(i))
    for(int j=y;j<=n;j+=lowbit(j))
    sum[i][j]+=s;
}
int query(int x,int y){
    int res=0;
    for(int i=x;i;i-=lowbit(i))
    for(int j=y;j;j-=lowbit(j))
    res+=sum[i][j];
    return res;
}
void solve(int ql,int qr,int l,int r){
    if(ql>qr) return;
    if(l==r){
        for(int i=ql;i<=qr;i++)
        ans[q[i].id]=l;
        return;
    }
    int mid=(l+r)>>1,l1=0,l2=0;
    for(;now<mid;) now++,add(a[now].x,a[now].y,1);
    for(;now>mid;now--) add(a[now].x,a[now].y,-1); 
    for(int i=ql;i<=qr;i++){
        int s=query(q[i].x2,q[i].y2)+query(q[i].x1-1,q[i].y1-1)-query(q[i].x1-1,q[i].y2)-query(q[i].x2,q[i].y1-1);
        if(q[i].s<=s) q1[++l1]=q[i];
        else q2[++l2]=q[i]; 
    }
    for(int i=1;i<=l1;i++)
    q[ql+i-1]=q1[i];
    for(int i=1;i<=l2;i++)
    q[ql+l1+i-1]=q2[i];
    if(l1) solve(ql,ql+l1-1,l,mid);
    if(l2) solve(ql+l1,qr,mid+1,r);
}
int main(){
    int x1,y1,x2,y2,x,y,z;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++){
        scanf("%d",&x);
        a[(i-1)*n+j]=(data){i,j,x}; 
    }
    sort(a+1,a+n*n+1,cmp);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&z);
        q[i]=(qry){x1,y1,x2,y2,z,i};
    }
    solve(1,m,1,n*n);
    for(int i=1;i<=m;i++)
    printf("%d\n",a[ans[i]].s);
}
/*
5 5
1 2 3 4 5
6 7 8 9 10
11 12 13 14 15
16 17 18 19 20
21 22 23 24 25
1 1 5 5 12
2 2 4 4 5
1 3 4 5 8
4 4 4 5 1
4 4 5 4 2
*/
/*
3 3 
101 233 1919810
666 999 6922
114 514 364364
1 1 2 2 3
1 3 3 3 2
3 3 3 3 1
*/