1. 程式人生 > >線段樹區間更新lazy

線段樹區間更新lazy

Color the ball

線段樹區間更新lazy

N個氣球排成一排,從左到右依次編號為1,2,3…N.每次給定2個整數a b(a <= b),lele便為騎上他的“小飛鴿"牌電動車從氣球a開始到氣球b依次給每個氣球塗一次顏色。但是N次以後lele已經忘記了第I個氣球已經塗過幾次顏色了,你能幫他算出每個氣球被塗過幾次顏色嗎?
Input
每個測試例項第一行為一個整數N,(N <= 100000).接下來的N行,每行包括2個整數a b(1 <= a <= b <= N)。
當N = 0,輸入結束。
Output
每個測試例項輸出一行,包括N個整數,第I個數代表第I個氣球總共被塗色的次數。
Sample Input
3
1 1
2 2
3 3
3
1 1
1 2
1 3
0
Sample Output
1 1 1
3 2 1

#include<cstdio>
using namespace std;
const int maxn=100005;
struct node{
	int l,r,sum,lazy;
}tr[maxn<<2];
void pushup(int m){
	tr[m].sum=tr[m<<1].sum+tr[m<<1|1].sum;
}
void build(int m,int l,int r){
	tr[m].l=l;
	tr[m].r=r;
	tr[m].lazy=0;
	if(l==r){
		//scanf("%d",&tr[m].sum);       
tr[m].sum=0; return ; } int mid=(l+r)>>1; build(m<<1,l,mid); build(m<<1|1,mid+1,r); pushup(m); } void pushdown(int m){ if(tr[m].lazy){ tr[m<<1].lazy+=tr[m].lazy; tr[m<<1|1].lazy+=tr[m].lazy; tr[m<<1].sum+=tr[m].lazy*(tr[m<<1].r-tr[m<<1]
.l+1); tr[m<<1|1].sum+=tr[m].lazy*(tr[m<<1|1].r-tr[m<<1|1].l+1); tr[m].lazy=0; } } void updata(int m,int l,int r,int val){ if(tr[m].l==l&&tr[m].r==r){ tr[m].sum+=val*(r-l+1); tr[m].lazy+=val; return ; } pushdown(m); int mid=(tr[m].l+tr[m].r)>>1; if(r<=mid) updata(m<<1,l,r,val); else if(l>mid) updata(m<<1|1,l,r,val); else{ updata(m<<1,l,mid,val); updata(m<<1|1,mid+1,r,val); } pushup(m); } int query(int m,int l,int r){ if(tr[m].l==l&&tr[m].r==r){ return tr[m].sum; } pushdown(m); int mid=(tr[m].l+tr[m].r)>>1; int temp; if(r<=mid) temp=query(m<<1,l,r); else if(l>mid) temp=query(m<<1|1,l,r); else temp=query(m<<1,l,mid)+query(m<<1|1,mid+1,r); return temp; } int main(){ int n,a,b; while(scanf("%d",&n)&&n){ build(1,1,n); for(int i=0;i<n;i++){ scanf("%d%d",&a,&b); updata(1,a,b,1); } for(int i=1;i<=n;i++){ if(i>1) printf(" "); printf("%d",query(1,i,i)); } printf("\n"); } return 0; }