hdu 4614 Vases and Flowers(线段树区间更新+二分)

题意:

给你N个花瓶,花瓶的编号是从 0 到 N - 1,初始状态花瓶是空的,每个花瓶最多只能插一朵花。

然后有2个操作:
操作1 a b,往在a位置后面(包括a)插b朵花,输出插入的首位置和末位置。
操作2 a b,输出区间 [a,b] 范围内的花的数量,然后全部清空。

解析:

很显然这是一道线段树的题目。区间更新,区间求和,这些基本的操作线段树都可以 O(log(n)) 的时间范围内完成。

操作2,很显然就是线段树的区间求和,求出[a , b]范围内的花朵的数量,然后区间更新,将整个区间全部变成0。

操作1,这里我们首先需要找出他的首位置和末位置,因为一个区间内的0的个数是单调非递减的,所以需要二分极大值极小化求0的位置。

首先我们二分他的首位置, ql=aqr=n1 ,在这个区间内二分,找出第一个0的位置,那就是该操作的首位置 first

然后再二分他的末位置, ql=first,r=n1 ,找到第c个0,就是该操作的末位置 last ,然后区间更新 [first,last] 全部置为1。

my code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define ls (o<<1)
#define rs (o<<1|1)
#define lson ls, L, M
#define rson rs, M+1, R
#define MID (L + R) >> 1
#define LEN(L, R) ((R) - (L) + 1)
using namespace std;
const int N = (int)5e4 + 10;
int n, m;
int cov[N<<2], sumv[N<<2];

void build(int o, int L, int R) {
    cov[o] = sumv[o] = 0;
    if(L == R) return ;
    int M = MID;
    build(lson);
    build(rson);
}

inline void pushUp(int o) {
    sumv[o] = sumv[ls] + sumv[rs];
}

void pushDown(int o, int L, int R) {
    if(cov[o] != -1) {
        int M = MID;
        cov[ls] = cov[rs] = cov[o];
        sumv[ls] = cov[o] * LEN(L, M);
        sumv[rs] = cov[o] * LEN(M+1, R);
        cov[o] = -1;
    }
}

void modify(int o, int L, int R, int ql, int qr, int val) {
    if(ql <= L && R <= qr) {
        cov[o] = val;
        sumv[o] = LEN(L, R) * val;
        return ;
    }
    pushDown(o, L, R);
    int M = MID;
    if(ql <= M) modify(lson, ql, qr, val);
    if(qr > M) modify(rson, ql, qr, val);
    pushUp(o);
}

int query(int o, int L, int R, int ql, int qr) {
    if(ql <= L && R <= qr) return sumv[o];
    pushDown(o, L, R);
    int M = MID, ret = 0;
    if(ql <= M) ret += query(lson, ql, qr);
    if(qr > M) ret += query(rson, ql, qr);
    return ret;
}

int query0(int ql, int qr) {
    return LEN(ql, qr) - query(1, 0, n-1, ql, qr);
}

int search(int start, int end, int cnt) {
    int L = start, R = end;
    while(L < R) {
        int M = MID;
        int tmp = query0(start, M);
        if(tmp < cnt) L = M + 1;
        else R = M;
    }
    return L;
}

int main() {
    int op, A, B;
    int ql, qr;

    int T;
    scanf("%d", &T);
    while(T--) {
        scanf("%d%d", &n, &m);
        build(1, 0, n-1);
        while(m--) {
            scanf("%d%d%d", &op, &A, &B);
            if(op == 1) {
                ql = A, qr = n-1;
                int zero = query0(ql, qr);
                if(zero == 0) {
                    puts("Can not put any one.");
                }else {
                    int first = search(ql, qr, 1);
                    int last = search(first, qr, min(zero, B));
                    printf("%d %d\n", first, last);
                    modify(1, 0, n-1, first, last, 1);
                }
            }else {
                ql = A, qr = B;
                printf("%d\n", query(1, 0, n-1, ql, qr));
                modify(1, 0, n-1, ql, qr, 0);
            }
        }
        puts("");
    }
    return 0;
}

你可能感兴趣的:(HDU,4614)