hdu 1556 树状数组(区间更新,单点查询)

Problem 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
0

Sample Output
1 1 1
3 2 1

题解:

在单点更新里面,树状数组求和操作代表区间的和,而在区间更新中,树状数组求和操作代表单个元素的变化量。

代码:

#include <bits/stdc++.h>
using namespace std;

const int maxn = 100000+100;

int bit[maxn],n;

int sum(int i)
{
    int s = 0;
    while(i>0)
    {
        s += bit[i];
        i -= i&-i;
    }
    return s;
}

void add(int i,int x)
{
    while(i<=n)
    {
        bit[i] += x;
        i += i&-i;
    }
}

int main()
{
    int a,b;
    while(~scanf("%d",&n)&&n)
    {
        memset(bit,0,sizeof(bit));
        for(int i=0;i<n;i++)
        {
            cin>>a>>b;
            add(a,1);
            add(b+1,-1);
        }
        for(int i=1;i<n;i++)
        {
           printf("%d ",sum(i));
        }
        //printf("%d\n",sum(5));
        printf("%d\n",sum(n));
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/q451792269/article/details/79561758
今日推荐