HDU 1556 Color the ball 【树状数组】

心悦灵溪

于 2019-08-02 17:31:19 发布

阅读量161

点赞数 1

分类专栏：树状数组

本文链接：https://blog.csdn.net/weixin_43847416/article/details/98215480

版权

树状数组专栏收录该内容

7 篇文章 0 订阅

订阅专栏

传送门：HDU 1556

Color the ball Time Limit: 9000/3000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

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

题解：
其实就是让我们求每个气球被涂色的次数，那么我们更新时会影响的就是超出区间的部分，我们把超出部分剪掉就行了。

AC代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int N=110000;
ll a[N],c[N];
int n;
int lowbit(int x)
{
  return x&(-x);
}
void update(int x,int k)
{
  while(x<=n)
  {
    c[x]+=k;
    x+=lowbit(x);
  }
}
ll getsum(int x)
{
  ll ans=0;
  for(int i=x;i>0;i-=lowbit(i))
  {
    ans+=c[i];
  }
  return ans;
}
int main()
{
  int l,r;
  while(scanf("%d",&n)&&n)
  {
    memset(a,0,sizeof(a));
    memset(c,0,sizeof(c));
    for(int i=0;i<n;i++)
    {
      scanf("%d%d",&l,&r);
      update(l,1);
      update(r+1,-1);
    }
    for(int i=1;i<=n;i++)
    {
      printf("%I64d%c",getsum(i),i==n?'\n':' ');
    }
  }
  return 0;
}