Description
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,输入结束。
当N = 0,输入结束。
Output
每个测试实例输出一行,包括N个整数,第I个数代表第I个气球总共被涂色的次数。
这题是区间修改,单点查询,可以用线段树或树状数组处理。
线段树做的核心代码:
int sum[N],add[N],ans[N]; void PushUp(int rt){ sum[rt]=sum[rt<<1]+sum[rt<<1|1]; } void PushDown(int rt,int m){ if (add[rt]!=0){ sum[rt<<1]+=add[rt]*(m-(m>>1)); sum[rt<<1|1]+=add[rt]*(m>>1); add[rt<<1]+=add[rt]; add[rt<<1|1]+=add[rt]; add[rt]=0; } } void update(int L,int R,int c,int l,int r,int rt){ if (L<=l&&R>=r){ add[rt]+=c; sum[rt]+=(l-r+1)*c; return; } int m=(l+r)>>1; PushDown(rt,r-l+1); if (L<=m) update(L,R,c,l,m,rt<<1); if (R>m) update(L,R,c,m+1,r,rt<<1|1); PushUp(rt); } void solve(int l,int r,int rt){ if (l==r){ ans[l]=sum[rt]; return; } int m=(l+r)>>1; PushDown(rt,r-l+1); solve(l,m,rt<<1); solve(m+1,r,rt<<1|1); } void Run(){ int n,a,b; while (~scanf("%d",&n)&&n){ memset(sum,0,sizeof(sum)); memset(add,0,sizeof(add)); for (int i=1;i<=n;i++){ scanf("%d %d",&a,&b); update(a,b,1,1,n,1); } solve(1,n,1); printf("%d",ans[1]); for (int i=2;i<=n;i++) printf(" %d",ans[i]); puts(""); } }
树状数组做的核心代码:
int num[N]; int tree[N]; int lowbit(int x){ return x&(-x); } void update(int x,int add){ while (x>0){ tree[x]+=add; x-=lowbit(x); } } int query(int x,int n){ int sum=0; while (x<=n){ sum+=tree[x]; x+=lowbit(x); } return sum; } void Run(int n){ memset(tree,0,sizeof(tree)); int a,b; for (int i=1;i<=n;i++){ scanf("%d %d",&a,&b); update(a-1,-1); update(b,1); } printf("%d",query(1,n)); for (int i=2;i<=n;i++) printf(" %d",query(i,n)); puts(""); }