欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-04-09 18:17:58
题意 题目链接 给出一个矩形,每个点都有一些值,每次询问一个子矩阵最少需要拿几个数才能构成给出的值 Sol 这题是真坑啊。。 首先出题人强行把两个题拼到了一起, 对于前$50 \%$的数据,考虑二分答案。 用$f[i][j][k]$表示从$(1, 1)$到$(i, j) >= k$的个数,$g[i] ......

题意

题目链接

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

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;
}
/*
*/