Poj 3667 线段树区间合并 （模板）

题目链接：http://poj.org/problem?id=3667

别人的分析：http://www.cnblogs.com/scau20110726/archive/2013/05/07/3065418.html

#include <iostream>
#include<algorithm>
#include<cstdio>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

using namespace std;
const int maxn=2e5;
int msum[maxn<<2],lsum[maxn<<2],rsum[maxn<<2];
int cover[maxn<<2];
void Pushup(int rt,int len)
{
lsum[rt]=lsum[rt<<1];
rsum[rt]=rsum[rt<<1|1];
if(lsum[rt]==(len-(len>>1)))lsum[rt]+=lsum[rt<<1|1];
if(rsum[rt]==(len>>1))rsum[rt]+=rsum[rt<<1];
msum[rt]=max(msum[rt<<1],max(msum[rt<<1|1],rsum[rt<<1]+lsum[rt<<1|1]));
}
void Pushdown(int rt,int len)
{
if(cover[rt]!=-1)
{
cover[rt<<1]=cover[rt<<1|1]=cover[rt];
msum[rt<<1]=lsum[rt<<1]=rsum[rt<<1]=cover[rt]?len-(len>>1):0;
msum[rt<<1|1]=lsum[rt<<1|1]=rsum[rt<<1|1]=cover[rt]?(len>>1):0;
cover[rt]=-1;
}
}
void build(int l,int r,int rt)
{
msum[rt]=lsum[rt]=rsum[rt]=r-l+1;
cover[rt]=-1;
if(l==r)return;
int m=(l+r)>>1;
build(lson);
build(rson);

}
void updata(int L,int R,int c,int l,int r,int rt)
{
if(L<=l&&R>=r)
{
msum[rt]=lsum[rt]=rsum[rt]=c?r-l+1:0;
cover[rt]=c;
return;
}
Pushdown(rt,r-l+1);
int m=(l+r)>>1;
if(L<=m)updata(L,R,c,lson);
if(R>m)updata(L,R,c,rson);
Pushup(rt,r-l+1);
}
int query(int w,int l,int r,int rt)
{
if(l==r)return l;
Pushdown(rt,r-l+1);
int m=(l+r)>>1;
if(msum[rt<<1]>=w)return query(w,lson);
if((rsum[rt<<1]+lsum[rt<<1|1])>=w)return m-rsum[rt<<1]+1;
else return query(w,rson);
}
int main()
{
int t,k,n,a,b,c,i;
while(~scanf("%d %d",&n,&k)&&n)
{
build(1,n,1);
while(k--)
{
scanf("%d",&a);
if(a==1)
{
scanf("%d",&b);
if(msum[1]<b)printf("0\n");
else {
int s=query(b,1,n,1);
printf("%d\n",s);
updata(s,b+s-1,0,1,n,1);
}
}
else
{
scanf("%d %d",&b,&c);
updata(b,c+b-1,1,1,n,1);
}
}
}
return 0;
}