总结:这题主要难在二分查找上,没怎么接触过,所以这次也略微的学了一下,线段树维护的是一段区间有多少个空瓶子,op = 1,的时候,要先统计一下从x ~ n有多少个空瓶子,如果为0,那么就是一个也不能放了,如果不为0,那么就要查找存放的第一个空瓶子和存放最后一个空瓶子所在的位置,那个二分查找自行模拟一变,就理解了。接下来就是将空瓶子插满了,此时的插满,是空瓶子数量减少。如果op = 2,直接用长度减空瓶子的数量就可以得到要清除的了,那么这时候空瓶子的数量就增加,也是通过线段树维护。通过这题我更加清晰的了解了关于线段树往上操作,与住下操作的具体深意。
#include <stdio.h>
#include <vector>
#include <string.h>
#include <algorithm>
#define mid (start+end)/2
#define ls (2*rt)
#define rs ((2*rt)+1)
#define len (end-start+1)
const int N = 5e4+5;
using namespace std;
int tree[N<<2],lazy[N<<2];
int n,m;
void build(int rt,int start,int end){
lazy[rt] = -1;
if(start == end){
tree[rt] = 1;
return;
}
build(ls,start,mid);
build(rs,mid+1,end);
tree[rt] = tree[ls]+tree[rs];
}
void push_down(int rt,int k){
if(lazy[rt] != -1){
lazy[ls] = lazy[rs] = lazy[rt];
tree[ls] = (k-k/2)*lazy[rt];
tree[rs] = k/2*lazy[rt];
lazy[rt] = -1;
}
}
int query(int rt,int start,int end,int l,int r){
if(start >= l && end <= r){
return tree[rt];
}
int ans = 0;
push_down(rt,len);
if(l <= mid) ans += query(ls,start,mid,l,r);
if(r > mid) ans += query(rs,mid+1,end,l,r);
return ans;
}
//二分查找,最关键,
//如果左侧的空瓶子数大于等于num,那么肯定就左侧,
//否则就在右侧
int bin_search(int x,int num){
int l = x,r = n;
int ans,m;
while(l <= r){
m = (l+r)/2;
if(query(1,1,n,x,m) >= num){
ans = m;
r = m-1;
}else l = m+1;
}
return ans;
}
void update(int rt,int start,int end,int l,int r,int op){
if(start >= l && end <= r){
tree[rt] = len*op;
lazy[rt] = op;
return;
}
if(l <= mid) update(ls,start,mid,l,r,op);
if(r > mid) update(rs,mid+1,end,l,r,op);
tree[rt] = tree[ls]+tree[rs];
}
void solve(){
scanf("%d%d",&n,&m);
int op,x,y;
build(1,1,n);
while(m--){
scanf("%d%d%d",&op,&x,&y);
if(op == 1){
int cnt = query(1,1,n,x+1,n);//统计从x到n有多少个空位子
if(cnt == 0){
printf("Can not put any one.\n");
}else {
int l = bin_search(x+1,1);
int r = bin_search(x+1,min(y,cnt));
update(1,1,n,l,r,0);
printf("%d %d\n",l-1,r-1);
}
}else {
x++,y++;
printf("%d\n",y-x+1-query(1,1,n,x,y));
update(1,1,n,x,y,1);
}
}
}
int main(){
int t;
scanf("%d",&t);
while(t--){
solve();
printf("\n");
}
return 0;
}