題目鏈接

BZOJ2738

題解

將矩陣中的位置取出來按權值排序
直接整體二分 + 二維BIT即可

#include<algorithm>
#include<iostream>
#include<cstdio>
#define LL long long int
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define lbt(x) (x & -x)
using namespace std;
const int maxn = 300005,maxm = 100005,INF = 0x3f3f3f3f;
inline int read(){
    int out = 0,flag = 1; char c = getchar();
    while (c < 48 || c > 57){if (c == '-') flag = 0; c = getchar();}
    while (c >= 48 && c <= 57){out = (out << 1) + (out << 3) + c - 48; c = getchar();}
    return flag ? out : -out;
}
struct node{int x,y,v;}e[maxn];
struct Que{int x1,y1,x2,y2,k,id;}q[maxn],t[maxn];
int b[maxn],bi,N,Q,n,tot,ans[maxn],s[505][505];
void add(int x,int y,int v){
    for (int i = x; i <= n; i += lbt(i))
        for (int j = y; j <= n; j += lbt(j))
            s[i][j] += v;
}
int query(int x,int y){
    int re = 0;
    for (int i = x; i; i -= lbt(i))
        for (int j = y; j; j -= lbt(j))
            re += s[i][j];
    return re;
}
int sum(int x1,int y1,int x2,int y2){
    return query(x2,y2) - query(x2,y1 - 1) - query(x1 - 1,y2) + query(x1 - 1,y1 - 1);
}
inline bool operator <(const node& a,const node& b){
    return a.v < b.v;
}
void solve(int l,int r,int L,int R){
    if (L > R) return;
    if (l == r){
        for (int i = L; i <= R; i++) ans[q[i].id] = e[l].v;
        return;
    }
    int mid = l + r >> 1,li = L,ri = R,v;
    for (int i = l; i <= mid; i++) add(e[i].x,e[i].y,1);
    for (int i = L; i <= R; i++){
        v = sum(q[i].x1,q[i].y1,q[i].x2,q[i].y2);
        if (v >= q[i].k) t[li++] = q[i];
        else q[i].k -= v,t[ri--] = q[i];
    }
    for (int i = L; i <= R; i++) q[i] = t[i];
    for (int i = l; i <= mid; i++) add(e[i].x,e[i].y,-1);
    solve(l,mid,L,li - 1); solve(mid + 1,r,ri + 1,R);
}
int main(){
    n = read(); Q = read();
    REP(i,n) REP(j,n) e[++N] = (node){i,j,b[++bi] = read()};
    REP(i,Q) q[i].x1 = read(),q[i].y1 = read(),q[i].x2 = read(),q[i].y2 = read(),q[i].k = read(),q[i].id = i;
    sort(b + 1,b + 1 + bi); tot = 1;
    for (int i = 2; i <= bi; i++) if (b[i] != b[tot]) b[++tot] = b[i];
    REP(i,N) e[i].v = lower_bound(b + 1,b + 1 + tot,e[i].v) - b;
    sort(e + 1,e + 1 + N);
    solve(1,N,1,Q);
    REP(i,Q) printf("%d\n",b[ans[i]]);
    return 0;
}
 

BZOJ2738 矩陣乘法 【整體二分 + BIT】