HDU 1556 Color the ball

这题用线段树的话简直就是一个水题。。不过刚学树状数组,要用一下。

题意:每次给你a,b,表明a~b之间涂色,然后最后一次输出每个气球被涂色的次数。

要用树状数组就要考虑怎么转化为前缀和问题,这题可以这样做:每次输入a,b,令A[a] = 1,A[b+1] = -1; 然后更新和,查询的时候容易知:a~b之间都被涂了一次,求前缀和结果也为一次,多次插入a,b,性质不变,插入后即可直接输出。复杂度很小,这也是树状数组的好处。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 100100

int c[N];
int n;

int lowbit(int k)
{
    return k&(-k);
}

void insert(int k,int num)
{
    while(k<=n)
    {
        c[k] += num;
        k += lowbit(k);
    }
}

int getsum(int k)
{
    int sum = 0;
    while(k>0)
    {
        sum += c[k];
        k -= lowbit(k);
    }
    return sum;
}

void init()
{
    int a,b,i;
    memset(c,0,sizeof(c));
    for(i=1;i<=n;i++)
    {
        scanf("%d%d",&a,&b);
        insert(a,1);
        insert(b+1,-1);
    }
}

int main()
{
    int j;
    while(cin>>n && n)
    {
        init();
        for(j=1;j<n;j++)
        {
            cout<<getsum(j)<<" ";
        }
        cout<<getsum(j)<<endl;
    }
    return 0;
}
View Code

代码参考了:http://blog.csdn.net/lulipeng_cpp/article/details/7816527

原文地址:https://www.cnblogs.com/whatbeg/p/3525561.html