洛谷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 = 1001, inf = 1e9 + 7, mod = 998244353, max = 5500002;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}
int n, m, q;
namespace subtask1 {
    int a[202][202], f[202][202][1002], g[202][202][1002];
    //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 - 1][r2][k] - g[l2][r1 - 1][k] + g[l1 - 1][r1 - 1][k];
    }
    int query(int k, int l1, int r1, int l2, int r2) {
        return f[l2][r2][k] - f[l1 - 1][r2][k] - f[l2][r1 - 1][k] + f[l1 - 1][r1 - 1][k];
    }
    void work() {
        for(int i = 1; i <= n; i++) {
            for(int j = 1; j <= m; j++) {
                a[i][j] = read();
                for(int k = 0; k <= 1000; k++) {
                    f[i][j][k] = f[i - 1][j][k] + f[i][j - 1][k] - f[i - 1][j - 1][k] + (a[i][j] >= k);
                    g[i][j][k] = g[i - 1][j][k] + g[i][j - 1][k] - g[i - 1][j - 1][k] + (a[i][j] >= k) * a[i][j];
                }
            }
        }
        while(q--) {
            int l1 = read(), r1 = read(), l2 = read(), r2 = read(), h = read();
            int l = 0, r =  1000, ans = -1;
            while(l <= r) {
                int mid = l + r >> 1;
                int x = get(mid, l1, r1, l2, r2, h);
                if(x >= h) ans = query(mid, l1, r1, l2, r2) - (x - h) / mid, l = mid + 1;
                else r = mid - 1;
            }
            if(ans == -1) {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 = 0;
    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) >> 1;
        if(pos <= mid) insert(ls[k], ls[p], l, mid, pos);
        else insert(rs[k], rs[p], mid + 1, r, pos);
    }
    int query(int x, int y, int l, int r, int val) {
        if(l == r) return (val + l - 1) / l;
        int mid = l + r >> 1;
        int rsum = sum[rs[y]]- sum[rs[x]];
        if(rsum >= val) return query(rs[x], rs[y], mid + 1, r, val);
        else return query(ls[x], ls[y], l, mid, val - rsum) + siz[rs[y]] - siz[rs[x]];
    }
    void work() {
        for(int i = 1; i <= m; i++) a[i] = read(), insert(rt[i], rt[i - 1], 1, 1000, a[i]);
        while(q--) {
            int l1 = read(), x = read(), l2 = read(), y = read(), h = read();
            if(l1 != 1 || l2 != 1) {puts("poor qlw"); continue;}
            if(sum[rt[y]] - sum[rt[x - 1]] < h) {puts("poor qlw"); continue;}
            int ans = query(rt[x - 1], rt[y], 1, 1000, h);
            printf("%d\n", ans);
        }
    }
}
main() {
  // freopen("a.in", "r", stdin);
  //  freopen("c.out", "w", stdout);
    n = read(); m = read(); q = read();
    if(n != 1) subtask1::work();
    else subtask2::work();
    return 0;
}
/*
*/