【bzoj 2738】矩阵乘法

传送门~

解题思路

整体二分。将一个询问拆成四个，二分答案x，查询每块中有多少比x小的数以验证答案。
代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<string>
#include<cstdlib>
using namespace std;
struct zhen{
    int xx,yy,u;
    bool operator < (const zhen &p) const{
         return u<p.u;
    }
}f[250005];
struct ldx{
    int x1,y1,x2,y2,num,k;
    void read(){
        scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&k);
    }
}s[60005],tmp[60005];
int a[505][505];
int ans[60005];
int n,m,lo=0,now=0;
void add(int x,int y,int num){
    for(int i=x;i<=n;i+=i&(-i))
    for(int j=y;j<=n;j+=j&(-j))
    a[i][j]+=num;
}
int ch(int x,int y){
    int ret=0;
    for(int i=x;i>0;i-=i&(-i))
    for(int j=y;j>0;j-=j&(-j))
    ret+=a[i][j];
    return ret;
}
void ZTF(int x,int y,int l,int r){
    int mid=(x+y)>>1;
    if(x>y || l>r) return ;
    while(now<lo && f[now+1].u<=mid){
        now++;
        add(f[now].xx,f[now].yy,1);
    }
    while(now>0 && f[now].u>mid){
        add(f[now].xx,f[now].yy,-1);
        now--;
    }
    int _l=l,_r=r;
    for(int i=l;i<=r;i++){
        int r1=ch(s[i].x2,s[i].y2)+ch(s[i].x1-1,s[i].y1-1)-ch(s[i].x2,s[i].y1-1)-ch(s[i].x1-1,s[i].y2);
        if(r1>=s[i].k){
            ans[s[i].num]=mid;
            tmp[_l++]=s[i];
        }
        else tmp[_r--]=s[i];
    }
    for(int i=l;i<=r;i++) s[i]=tmp[i];

    ZTF(x,mid-1,l,_l-1);
    ZTF(mid+1,y,_r+1,r);
}
int main(){
    int rmx;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++){
        lo++;scanf("%d",&f[lo].u);
        f[lo].xx=i;f[lo].yy=j;
        rmx=max(rmx,f[lo].u);
    }
    sort(f+1,f+lo+1);
    for(int i=1;i<=m;i++) {
        s[i].read();s[i].num=i;
    }
    ZTF(0,rmx,1,m);
    for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
    return 0;
}