题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4614
题目大意:n个花,m次操作,每次操作都可以将一个区间的花都拔掉,输出拔掉的个数。或者插入k个花到花瓶中,输出插入的起点和终点。
思路:每次插花的时候,找到第一个可以插入的位置,后面再找数量<=k的位置。将整个区间更新为0.把花的时候,直接将整个区间更新为1即可。找对应的插花区间的时候,可以用二分查找。
ACCode:
#include<stdlib.h>
#include<string.h>
#include<stdio.h>
#include<time.h>
#include<math.h>
// srand(unsigned)time(NULL));rand();
#include<map>
#include<set>
#include<deque>
#include<queue>
#include<stack>
#include<bitset>
#include<string>
#include<fstream>
#include<iostream>
#include<algorithm>
#define ll long long
#define Pii pair<int,int>
#define clean(a,b) memset(a,b,sizeof(a))
using namespace std;
const int MAXN=5e6+10;
const int INF32=0x3f3f3f3f;
const ll INF64=0x3f3f3f3f3f3f3f3f;
const ll MOD=10007;
const double PI=acos(-1.0);
const double EPS=1.0e-8;
//unsigned register
// ios::sync_with_stdio(false)
struct SegTree{
struct Node{
int l,r,len;
int Sum;
int lazy;
};
Node Tree[MAXN<<2];
void PushUp(int rt){
Tree[rt].Sum=Tree[rt<<1].Sum+Tree[rt<<1|1].Sum;
}
void PushDown(int rt){
if(Tree[rt].lazy!=-1){
int val=Tree[rt].lazy;
Tree[rt<<1].Sum=Tree[rt<<1].len*val;
Tree[rt<<1|1].Sum=Tree[rt<<1|1].len*val;
Tree[rt<<1].lazy=Tree[rt<<1|1].lazy=val;
Tree[rt].lazy=-1;
}
}
void Build(int l,int r,int rt){
Tree[rt].l=l;Tree[rt].r=r;Tree[rt].len=r-l+1;
Tree[rt].lazy=-1;
if(l==r){
Tree[rt].Sum=1;
return ;
}
int mid=(l+r)>>1;
Build(l,mid,rt<<1);Build(mid+1,r,rt<<1|1);
PushUp(rt);
}
void Update(int ql,int qr,int val,int rt){
if(ql<=Tree[rt].l&&Tree[rt].r<=qr){
Tree[rt].Sum=val*Tree[rt].len;
Tree[rt].lazy=val;
return ;
}PushDown(rt);
if(qr<=Tree[rt<<1].r) Update(ql,qr,val,rt<<1);
else if(ql>=Tree[rt<<1|1].l) Update(ql,qr,val,rt<<1|1);
else{
Update(ql,qr,val,rt<<1);
Update(ql,qr,val,rt<<1|1);
}PushUp(rt);
}
int Query(int ql,int qr,int rt){
if(ql>qr) return 0;
//printf("ql=%d qr=%d rtl=%d rtr=%d\n",ql,qr,Tree[rt].l,Tree[rt].r);
if(ql<=Tree[rt].l&&Tree[rt].r<=qr) return Tree[rt].Sum;
PushDown(rt);
int Ans;
if(qr<=Tree[rt<<1].r) Ans=Query(ql,qr,rt<<1);
else if(ql>=Tree[rt<<1|1].l) Ans=Query(ql,qr,rt<<1|1);
else Ans=Query(ql,qr,rt<<1)+Query(ql,qr,rt<<1|1);
PushUp(rt);
return Ans;
}
int GetL(int l,int r,int K,int mid=0){
while(l<=r){
mid=(l+r)>>1;
if(Query(l,mid,1)>0) r=mid-1;
else l=mid+1;
}return l;
}
int GetLastR(int l,int r,int mid=0){
while(l<=r){
mid=(l+r)>>1;
if(Query(mid,r,1)>0) l=mid+1;
else r=mid-1;
}return r;
}
int GetMidR(int l,int r,int K,int mid=0){
int str=l;
while(l<=r){
mid=(l+r)>>1;
//cout<<l<<" "<<mid<<" "<<r<<endl;
if(Query(str,mid,1)>=K) r=mid-1;
else l=mid+1;
}return l;
}
void Show(int rt){
printf("l=%d r=%d Sum=%d\n",Tree[rt].l,Tree[rt].r,Tree[rt].Sum);
if(Tree[rt].l==Tree[rt].r) return ;
PushDown(rt);
Show(rt<<1);Show(rt<<1|1);
PushUp(rt);
}
};
SegTree Seg;
int n,m;
int main(){
int T;scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&m);
Seg.Build(1,n,1);//Seg.Show(1);
while(m--){
int opt,l,r;scanf("%d%d%d",&opt,&l,&r);
if(opt==1){//从l处插花r个
++l;
int Sum=Seg.Query(l,n,1);//可以插花
//printf("Sum=%d\n",Sum);
if(Sum==0||r==0){
printf("Can not put any one.\n");
continue;
}
int ql,qr;
ql=Seg.GetL(l,n,r);
//printf("ql=%d\n",ql);
if(r>Sum) qr=Seg.GetLastR(ql,n);//获得插花区间
else qr=Seg.GetMidR(ql,n,r);
//printf("qr=%d\n",qr);
Seg.Update(ql,qr,0,1);//更新插花区间
--ql;--qr;
printf("%d %d\n",ql,qr);
}
else{//消除[l,r]区间的花
++l;++r;
int Sum=Seg.Tree[1].Sum;
Seg.Update(l,r,1,1);
Sum=abs(Sum-Seg.Tree[1].Sum);
printf("%d\n",Sum);
}//Seg.Show(1);
}printf("\n");
}
}
/*
1
14 10
1 5 12
2 3 7
1 2 9
2 2 13
2 3 10
1 2 5
2 2 2
2 12 13
1 0 0
1 0 2
1
10 6
1 2 2
2 4 8
1 1 3
1 2 2
1 5 6
1 1 2
*/