HDU1556 Color the ball(线段树)

#include <iostream>
#include <cstdio>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn=100010;
int segs[maxn<<2],f[maxn<<2],fl,n;

void pushDown(int rt){
if(f[rt]){
segs[rt<<1]+=f[rt];
f[rt<<1]+=f[rt];
segs[rt<<1|1]+=f[rt];
f[rt<<1|1]+=f[rt];
f[rt]=0;
}
}

void update(int L,int R,int l,int r,int rt){
if(L<=l&&r<=R){
segs[rt]+=(r-l+1);
f[rt]+=1;
return;
}
int m=(l+r)>>1;
pushDown(rt);
if(L<=m)update(L,R,lson);
if(R>m)update(L,R,rson);
}

void print(int l,int r,int rt){
if(l==r){
if(fl++>0)printf(" ");
printf("%d",segs[rt]);
return;
}
int m=(l+r)>>1;
pushDown(rt);
print(lson),print(rson);
}

int main(){
while (scanf("%d",&n)&&n){
fl=0;
memset(segs,0,sizeof(segs));
memset(f,0,sizeof(f));
for (int i=0;i<n;++i)
{
int a,b;
scanf("%d%d",&a,&b);
update(a,b,1,n,1);
}
print(1,n,1);
printf("\n");
}
return 0;
}