HDU 1556 Color the ball（线段树|树状数组）

Color the ball

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 21795    Accepted Submission(s): 10552

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
  
  

 

Author

8600

 

Source

HDU 2006-12 Programming Contest 

 

中文题意不做解释了。

法一：

用的是线段树，就是区间更新和单点查询，就是单点查询每个都要查询到，感觉效率有点慢。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <map>
#include <algorithm>
#include <math.h>
using namespace std;
#define rt x<<1|1
#define lt x<<1
#define  LL long long
const int N = 100000*5;
int num[N];
int Add[N];
void pushdown(int x)
{
    Add[lt]+=Add[x];
    Add[rt]+=Add[x];
    num[lt]+=Add[x];
    num[rt]+=Add[x];
    Add[x]=0;
}
void add(int x,int l,int r,int ll,int rr)
{
    ///if(Add[x]) pushdown(x);
    if(ll<=l&&rr>=r)
    {
        num[x]++;
        Add[x]++;
    }
    else
    {
        int mid=(l+r)>>1;
        if(ll<=mid) add(lt,l,mid,ll,rr);
        if(mid<rr) add(rt,mid+1,r,ll,rr);
        num[x]=num[lt]+num[rt];
    }
}
int query(int x,int l,int r,int i)
{
    if(Add[x]) pushdown(x);
    int ans=0;
    if(l==r&&l==i)
    {
        ans+=num[x];
    }
    else
    {
        int mid=(l+r)>>1;
        if(i<=mid) ans+=query(lt,l,mid,i);
        if(mid<i) ans+=query(rt,mid+1,r,i);
    }
    return ans;
}
int main()
{
    int n;
    while(~scanf("%d",&n)&&n)
    {
        int l,r;
        memset(num,0,sizeof num);
        memset(Add,0,sizeof Add);
        for(int i=1;i<=n;i++)
        {
            scanf("%d %d",&l,&r);
            add(1,1,n,l,r);
        }
        for(int i=1;i<=n;i++)
        {
            if(i!=1) printf(" ");
            printf("%d",query(1,1,n,i));
        }
        printf("\n");
    }
    
}

法二：

树状数组，把问题当作单点更新，区间查询。

遇到左端点，左端点所在的+1，右端点+1所在的-1。查询点就直接变成了查询1-x区间上的值。这个思维就和法一完全不一样了。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <map>
#include <algorithm>
#include <math.h>
using namespace std;
#define rt x<<1|1
#define lt x<<1
#define  LL long long
const int N = 100000*5;
int n;
int num[N];
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int c)
{
    while(x<=n)
    {
        num[x]+=c;
        x+=lowbit(x);
    }
}
int query(int x)
{
    int ans=0;
    while(x>=1)
    {
        ans+=num[x];
        x-=lowbit(x);
    }
    return ans;
}
int main()
{
    while(~scanf("%d",&n)&&n)
    {
        memset(num,0,sizeof num);
        int l,r;
        for(int i=1;i<=n;i++)
        {
            scanf("%d %d",&l,&r);
            add(l,1);
            add(r+1,-1);
        }
        for(int i=1;i<=n;i++)
        {
            printf("%d",query(i));
            if(i==n) printf("\n");
            else printf(" ");
        }
    }
}