hdu1556题解

B - Color the ball
Time Limit:3000MS Memory Limit:32768KB 64bit IO Format:%I64d & %I64u
Submit

Status

Practice

HDU 1556
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，输入结束。

Output
每个测试实例输出一行，包括N个整数，第I个数代表第I个气球总共被涂色的次数。

Sample Input
3
1 1
2 2
3 3
3
1 1
1 2
1 3

线段树中比较麻烦的区间修改，在线段树的点修改中，我们找到需要的叶子节点并更新，然后递归回溯更新所有父亲节点，但是在区间更改中，我们最坏情况下需要更新所有的叶子节点，效率不高，有一种解决方案。

比如我要把区间（L，R）的所有节点加1，我不需要找到（L,R）的所有叶子节点，我只需要找到组成（L,R)的子区间，对子区间执行add操作。每次标记add时，需要回溯维护其父亲节点，查询时依然找到所有子区间，对所有结果求和，需要注意的是需要考虑父亲节点的影响。

#include <cstdio>#include <cstring>#include <iostream>using namespace std;const int maxnum=100000+5;int segSum[maxnum*4];int add[maxnum*4];void maintain(int node,int begin,int end){    int c1=node*2,c2=node*2+1;    segSum[node]=0;    if(end>begin)    {        segSum[node]=segSum[c1]+segSum[c2];    }    segSum[node]+=add[node]*(end-begin+1);}void update(int node,int begin,int end,int left,int right){    if(left<=begin && end<=right)    {        add[node]+=1; //cout<<"node="<<node<<endl;        maintain(node,begin,end);        return;    }    int m=(begin+end)/2;    if(left<=m)    {        update(node*2,begin,m,left,right);    }    if(right>m)    {        update(node*2+1,m+1,end,left,right);    }    maintain(node,begin,end);}int _sum;void query(int node,int begin,int end,int left,int right,int val){    if(left<=begin && right>=end)    {        _sum+=segSum[node]+val*(end-begin+1);return;    }    int m=(begin+end)/2;    if(left<=m)    {        query(node*2,begin,m,left,right,val+add[node]);    }    if(right>m)    {        query(node*2+1,m+1,end,left,right,val+add[node]);    }}int  main(){    int n;    while(scanf("%d",&n) && n)    {        memset(segSum,0,sizeof(segSum));        memset(add,0,sizeof(add));        int a,b;        for(int k=1;k<=n;k++)        {            scanf("%d%d",&a,&b);            update(1,1,n,a,b);        }        _sum=0;        int temp=0;        query(1,1,n,1,1,0);        for(int i=2;i<=n;i++)        {            printf("%d ",_sum-temp);            temp=_sum;            _sum=0;            query(1,1,n,1,i,0);        }        printf("%d\n",_sum-temp);    }    return 0;}