给你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=a,qr=n−1 ,在这个区间内二分,找出第一个0的位置,那就是该操作的首位置 first 。
然后再二分他的末位置, ql=first,r=n−1 ,找到第c个0,就是该操作的末位置 last ,然后区间更新 [first,last] 全部置为1。
my code
#include
#include
#include
#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;
}