洛谷P2468 [SDOI2010]粟粟的书架(二分答案 前缀和 主席树)

题意

题目链接

给出一个矩形，每个点都有一些值，每次询问一个子矩阵最少需要拿几个数才能构成给出的值

Sol

这题是真坑啊。。

首先出题人强行把两个题拼到了一起，

对于前$50 \%$的数据，考虑二分答案。

用$f[i][j][k]$表示从$(1, 1)$到$(i, j) >= k$的个数,$g[i][j][k]$表示从$(1, 1)$到$(i, j) >= k$的和

这两个数组都可以递推出来

对于后$50 \%$，直接用主席树维护。。

mdzz死活有一个点RE。。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#define LL long long
// #define int long long
using namespace std;
const int MAXN = , INF = 1e9 + , mod = , MAX = ;
inline int read() {
    char c = getchar(); int x = , f = ;
    while(c < '' || c > '') {if(c == '-') f = -; c = getchar();}
    while(c >= '' && c <= '') x = x *  + c - '', c = getchar();
    return x * f;
}
int N, M, Q;
namespace SubTask1 {
    int a[][], f[][][], g[][][];
    //f[i][j][k] 从(1, 1)到(i, j) >= k的个数
    //g[i][j][k] 从(1, 1)到(i, j) >= k的和
    int Get(int k, int l1, int r1, int l2, int r2, int H) {
        return g[l2][r2][k] - g[l1 - ][r2][k] - g[l2][r1 - ][k] + g[l1 - ][r1 - ][k];
    }
    int Query(int k, int l1, int r1, int l2, int r2) {
        return f[l2][r2][k] - f[l1 - ][r2][k] - f[l2][r1 - ][k] + f[l1 - ][r1 - ][k];
    }
    void work() {
        for(int i = ; i <= N; i++) {
            for(int j = ; j <= M; j++) {
                a[i][j] = read();
                for(int k = ; k <= ; k++) {
                    f[i][j][k] = f[i - ][j][k] + f[i][j - ][k] - f[i - ][j - ][k] + (a[i][j] >= k);
                    g[i][j][k] = g[i - ][j][k] + g[i][j - ][k] - g[i - ][j - ][k] + (a[i][j] >= k) * a[i][j];
                }
            }
        }
        while(Q--) {
            int l1 = read(), r1 = read(), l2 = read(), r2 = read(), H = read();
            int l = , r =  , ans = -;
            while(l <= r) {
                int mid = l + r >> ;
                int X = Get(mid, l1, r1, l2, r2, H);
                if(X >= H) ans = Query(mid, l1, r1, l2, r2) - (X - H) / mid, l = mid + ;
                else r = mid - ;
            }
            if(ans == -) {puts("Poor QLW"); continue;}
            printf("%d\n", ans);
        }
    }
}
namespace SubTask2 {
    int a[MAX];
    int ls[MAX], rs[MAX], sum[MAX], siz[MAX], rt[MAX], tot = ;
    void Insert(int &k, int p, int l, int r, int pos) {
        k = ++tot;
        rs[k] = rs[p]; ls[k] = ls[p]; siz[k] = siz[p]; sum[k] = sum[p];
        siz[k]++; sum[k] += pos;
        if(l == r) return ;
        int mid = (l + r) >> ;
        if(pos <= mid) Insert(ls[k], ls[p], l, mid, pos);
        else Insert(rs[k], rs[p], mid + , r, pos);
    }
    int Query(int x, int y, int l, int r, int val) {
        if(l == r) return (val + l - ) / l;
        int mid = l + r >> ;
        int rsum = sum[rs[y]]- sum[rs[x]];
        if(rsum >= val) return Query(rs[x], rs[y], mid + , r, val);
        else return Query(ls[x], ls[y], l, mid, val - rsum) + siz[rs[y]] - siz[rs[x]];
    }
    void work() {
        for(int i = ; i <= M; i++) a[i] = read(), Insert(rt[i], rt[i - ], , , a[i]);
        while(Q--) {
            int l1 = read(), x = read(), l2 = read(), y = read(), H = read();
            if(l1 !=  || l2 != ) {puts("Poor QLW"); continue;}
            if(sum[rt[y]] - sum[rt[x - ]] < H) {puts("Poor QLW"); continue;}
            int ans = Query(rt[x - ], rt[y], , , H);
            printf("%d\n", ans);
        }
    }
}
main() {
  // freopen("a.in", "r", stdin);
  //  freopen("c.out", "w", stdout);
    N = read(); M = read(); Q = read();
    if(N != ) SubTask1::work();
    else SubTask2::work();
    return ;
}
/*
*/