poj3667(线段树区间合并)

链接：点击打开链接

题意：给出包含n个元素初始为0的数组，有两种操作，一种返回连续是0的长度为d的最左边的下标并区间更新成1，另一种将区间更新为0

代码：#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int siz=50005;
int lazy[siz<<2];
int sum[siz<<2],lsum[siz<<2],rsum[siz<<2];
void build(int l,int r,int rt){
lazy[rt]=-1;
sum[rt]=lsum[rt]=rsum[rt]=r-l+1;
if(l==r)
return;
int m=(l+r)>>1;
build(l,m,rt<<1);
build(m+1,r,rt<<1|1);
}
void pushup(int rt,int p){
lsum[rt]=lsum[rt<<1];
rsum[rt]=rsum[rt<<1|1];
if(lsum[rt<<1]==(p-(p>>1)))
lsum[rt]+=lsum[rt<<1|1];
if(rsum[rt<<1|1]==(p>>1))
rsum[rt]+=rsum[rt<<1]; //区间进行合并
sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
sum[rt]=max(sum[rt],rsum[rt<<1]+lsum[rt<<1|1]);
}
void pushdown(int rt,int p){
if(lazy[rt]!=-1){ //懒惰标记
if(lazy[rt]==0){
lazy[rt<<1]=lazy[rt<<1|1]=0;
sum[rt<<1]=lsum[rt<<1]=rsum[rt<<1]=p-(p>>1);
sum[rt<<1|1]=lsum[rt<<1|1]=rsum[rt<<1|1]=(p>>1);
}
else{
lazy[rt<<1]=lazy[rt<<1|1]=1;
sum[rt<<1]=lsum[rt<<1]=rsum[rt<<1]=0;
sum[rt<<1|1]=lsum[rt<<1|1]=rsum[rt<<1|1]=0;
}
lazy[rt]=-1;
}
}
void Change(int L,int R,int p,int l,int r,int rt){
if(L<=l&&r<=R){
lazy[rt]=p;
if(p==0)
sum[rt]=lsum[rt]=rsum[rt]=r-l+1;
else
sum[rt]=lsum[rt]=rsum[rt]=0;
return;
}
pushdown(rt,r-l+1);
int m=(l+r)>>1;
if(L<=m)
Change(L,R,p,l,m,rt<<1);
if(R>m)
Change(L,R,p,m+1,r,rt<<1|1);
pushup(rt,r-l+1);
}
int query(int p,int l,int r,int rt){
if(l==r)
return l;
pushdown(rt,r-l+1);
int m=(l+r)>>1;
if(sum[rt<<1]>=p)
return query(p,l,m,rt<<1);
else if(rsum[rt<<1]+lsum[rt<<1|1]>=p)
return m-rsum[rt<<1]+1;
else
return query(p,m+1,r,rt<<1|1);
}
int main(){ //sum[rt]表示rt这个节点最长连续0
int n,m,u,v,w,op; //lsum[rt]表示从rt区间左端点开始最长连续0
while(scanf("%d%d",&n,&m)!=EOF){ //rsum[rt]表示从rt区间右端点开始最长连续0
build(1,n,1);
while(m--){
scanf("%d",&op);
if(op==1){
scanf("%d",&u);
if(sum[1]<u)
puts("0");
else{
w=query(u,1,n,1);
printf("%d\n",w);
Change(w,w+u-1,1,1,n,1);
}
}
else{
scanf("%d%d",&u,&v);
Change(u,u+v-1,0,1,n,1);
}
}
}
return 0;
}