首页 > 代码库 > Color the ball 前缀和+思维

Color the ball 前缀和+思维

Color the ball

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

31 12 23 331 11 21 30

Sample Output

1 1 13 2 1


正常做法是用线段树。不过还有一个更简单巧妙的O(n)方法:当一段区间+1时,将区间首元素+1,尾元素的下一元素-1(区间[x,y]:a[x]++;b[y+1]--;),就等价于对这段区间做了处理。在最后输出时每一个元素都加上他前面一个元素的值(a[i]+=a[i-1];)即为最后的结果。

#include<stdio.h>#include<string.h>int a[100005];int main(){    int n,x,y,i;    while(scanf("%d",&n)&&n!=0){        memset(a,0,sizeof(a));        for(i=1;i<=n;i++){            scanf("%d%d",&x,&y);            a[x]++;            a[y+1]--;        }        printf("%d",a[1]);        for(i=2;i<=n;i++){            a[i]+=a[i-1];            printf(" %d",a[i]);        }        printf("\n");    }    return 0;}

 

 

Color the ball 前缀和+思维