【BZOJ1926】【SDOI2010】粟粟的书架(DP,主席树,二分)

Description

幸福幼儿园B29班的粟粟是一个聪明机灵、乖巧可爱的小朋友,她的爱好是画画和读书,尤其喜欢Thomas H. Cormen的文章。粟粟家中有一个R行C列的巨型书架,书架的每一个位置都摆有一本书,上数第i行、左数第j列摆放的书有Pi,j页厚。

粟粟每天除了读书之外,还有一件必不可少的工作就是摘苹果,她每天必须摘取一个指定的苹果。粟粟家果树上的苹果有的高、有的低,但无论如何凭粟粟自己的个头都难以摘到。不过她发现,如果在脚下放上几本书,就可以够着苹果;她同时注意到,对于第i天指定的那个苹果,只要她脚下放置书的总页数之和不低于Hi,就一定能够摘到。

由于书架内的书过多,父母担心粟粟一天内就把所有书看完而耽误了上幼儿园,于是每天只允许粟粟在一个特定区域内拿书。这个区域是一个矩形,第i天给定区域的左上角是上数第x1i行的左数第y1i本书,右下角是上数第x2i行的左数第y2i本书。换句话说,粟粟在这一天,只能在这﹙x2i-x1i+1﹚×﹙y2i-y1i+1﹚本书中挑选若干本垫在脚下,摘取苹果。

粟粟每次取书时都能及时放回原位,并且她的书架不会再撤下书目或换上新书,摘苹果的任务会一直持续M天。给出每本书籍的页数和每天的区域限制及采摘要求,请你告诉粟粟,她每天至少拿取多少本书,就可以摘到当天指定的苹果。

Constraints

对于10%的数据,满足R, C≤10;

对于20%的数据,满足R, C≤40;

对于50%的数据,满足R, C≤200,M≤200,000;

另有50%的数据,满足R=1,C≤500,000,M≤20,000;

对于100%的数据,满足1≤Pi,j≤1,000,1≤Hi≤2,000,000,000。

Solution

从数据范围可以看出题目可以看做两个子问题。

R,C200 R , C ≤ 200

fi,j,k f i , j , k 表示子矩阵 (1,1)(i,j) ( 1 , 1 ) ( i , j ) 中大于等于 k k 的数的和;
gi,j,k表示子矩阵 (1,1)(i,j) ( 1 , 1 ) ( i , j ) 中大于等于 k k 的个数。
然后二分所取的最小的数即可。

R=1

跟上面一样的二分思路,用主席树维护即可。

Source

/************************************************
 * Au: Hany01
 * Date: Feb 27th, 2018
 * Prob: [BZOJ1926][SDOI2010] 粟粟的书架
 * Email: hany01@foxmail.com
************************************************/

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<int, int> PII;
#define rep(i, j) for (register int i = 0, i##_end_ = (j); i < i##_end_; ++ i)
#define For(i, j, k) for (register int i = (j), i##_end_ = (k); i <= i##_end_; ++ i)
#define Fordown(i, j, k) for (register int i = (j), i##_end_ = (k); i >= i##_end_; -- i)
#define Set(a, b) memset(a, b, sizeof(a))
#define Cpy(a, b) memcpy(a, b, sizeof(a))
#define fir first
#define sec second
#define pb(a) push_back(a)
#define mp(a, b) make_pair(a, b)
#define ALL(a) (a).begin(), (a).end()
#define SZ(a) ((int)(a).size())
#define INF (0x3f3f3f3f)
#define INF1 (2139062143)
#define Mod (1000000007)
#define debug(...) fprintf(stderr, __VA_ARGS__)
#define y1 wozenmezhemecaia

template <typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }
template <typename T> inline bool chkmin(T &a, T b) { return b < a ? a = b, 1 : 0; }

inline int read()
{
    register int _, __; register char c_;
    for (_ = 0, __ = 1, c_ = getchar(); c_ < '0' || c_ > '9'; c_ = getchar()) if (c_ == '-') __ = -1;
    for ( ; c_ >= '0' && c_ <= '9'; c_ = getchar()) _ = (_ << 1) + (_ << 3) + (c_ ^ 48);
    return _ * __;
}

inline void File()
{
#ifdef hany01
    freopen("bzoj1926.in", "r", stdin);
    freopen("bzoj1926.out", "w", stdout);
#endif
}

const int maxn = 500005, maxm = 205, maxk = 1005;

int a[maxm][maxm], n, m, sum[maxm][maxm][maxk], num[maxm][maxm][maxk], q;

inline void Solve1()
{
    For(i, 1, n) For(j, 1, m) {
        a[i][j] = read();
        For(k, 1, a[i][j])
            sum[i][j][k] = sum[i - 1][j][k] + sum[i][j - 1][k] - sum[i - 1][j - 1][k] + a[i][j],
            num[i][j][k] = num[i - 1][j][k] + num[i][j - 1][k] - num[i - 1][j - 1][k] + 1;
        For(k, a[i][j] + 1, 1000)
            sum[i][j][k] = sum[i - 1][j][k] + sum[i][j - 1][k] - sum[i - 1][j - 1][k],
            num[i][j][k] = num[i - 1][j][k] + num[i][j - 1][k] - num[i - 1][j - 1][k];
    }
    static int x1, y1, x2, y2, h, l, r, mid, Sum, Num;
    while (q --) {
        x1 = read() - 1, y1 = read() - 1, x2 = read(), y2 = read(), h = read();
        l = 1, r = 1000;
        while (l < r) {
            mid = (l + r + 1) >> 1;
            if (sum[x2][y2][mid] + sum[x1][y1][mid] - sum[x1][y2][mid] - sum[x2][y1][mid] >= h) l = mid; else r = mid - 1;
        }
        Sum = sum[x2][y2][l] + sum[x1][y1][l] - sum[x1][y2][l] - sum[x2][y1][l];
        Num = num[x2][y2][l] + num[x1][y1][l] - num[x1][y2][l] - num[x2][y1][l];
        if (Sum >= h) printf("%d\n", Num - (Sum - h) / l); else puts("Poor QLW");
    }
}

struct President_Tree
{
    struct Node
    {
        int val, l, r, num;
    }tr[maxn * 20];
    int cnt;

#define mi ((l + r) >> 1)

    int update(int last, int l, int r, int value)
    {
        int now = ++ cnt;
        tr[now] = tr[last], ++ tr[now].num, tr[now].val += value;
        if (l < r) {
            if (value <= mi) tr[now].l = update(tr[last].l, l, mi, value);
            else tr[now].r = update(tr[last].r, mi + 1, r, value);
        }
        return now;
    }

    int query(int x, int y, int l, int r, int k)
    {
        if (l == r) return l * k;
        int num = tr[tr[y].l].num - tr[tr[x].l].num;
        if (num >= k) return query(tr[x].l, tr[y].l, l, mi, k);
        return tr[tr[y].l].val - tr[tr[x].l].val + query(tr[x].r, tr[y].r, mi + 1, r, k - num);
    }

}PT;

inline void Solve2()
{
    static int a[maxn], rt[maxn], l, r, mid, x, y, h, all, tmp;
    n = m;
    For(i, 1, n) rt[i] = PT.update(rt[i - 1], 1, 1000, a[i] = read());
    while (q --) {
        tmp = read(), x = read(), tmp = read(), y = read(), h = read();
        all = PT.query(rt[x - 1], rt[y], 1, 1000, y - x + 1);
        if (all < h) { puts("Poor QLW"); continue; }
        l = 1, r = y - x + 1;
        while (l < r) {
            mid = (l + r) >> 1;
            if (all - PT.query(rt[x - 1], rt[y], 1, 1000, y - x + 1 - mid) >= h) r = mid; else l = mid + 1;
        }
        printf("%d\n", l);
    }
}

int main()
{
    File();
    n = read(), m = read(), q = read();
    if (n > 1) Solve1(); else Solve2();
    return 0;
}
//一年将尽夜,万里未归人。
//    -- 戴叔伦《除夜宿石头驿》
©️2020 CSDN 皮肤主题: 撸撸猫 设计师:设计师小姐姐 返回首页