HDU1556：Color the ball(线段树区间更新单点求值)&&树状数组解法

Color the ball

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

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

 

Recommend

LL

 

Statistic |  Submit |  Discuss |  Note
Link

树状数组思路：直接利用求和函数求单点的值，区间更新只更新起点（i），在（j+1）处设置一个标记表示更新到 j 

AC code：

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

const int maxn=100000+50;
const int INF=0x3f3f3f3f;
int n;
int bit[maxn];

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(){
    while(~scanf("%d",&n),n){
        memset(bit,0,sizeof(bit));
        for(int k=0;k<n;k++){
            int i,j;
            scanf("%d%d",&i,&j);
            add(i,1);
            add(j+1,-1);
        }
        for(int i=1;i<n;i++) printf("%d ",sum(i));
        printf("%d\n",sum(n));
    }
    return 0;
}

线段树区间更新单点查询：

AC code：

#include<cstdio>
#include<algorithm>
#include<iostream>
#define LL long long
using namespace std;
const int MAXN=100000;
//int num[MAXN];

//线段树区间更新，区间和值查询

struct Node
{
    int l,r;//区间的左右端点
    LL sum;//区间上的和
    LL Inc;//区间增量的累加
}segTree[MAXN*3];
void Build(int i,int l,int r)
{
    segTree[i].l=l;
    segTree[i].r=r;
    segTree[i].Inc=0;
    if(l==r)
    {
        segTree[i].sum=0;
        return;
    }
    int mid=(l+r)>>1;
    Build(i<<1,l,mid);
    Build(i<<1|1,mid+1,r);
    segTree[i].sum=segTree[i<<1].sum+segTree[i<<1|1].sum;
}
void Add(int i,int a,int b,LL c)//在结点i的区间（a,b）上增加c
{
    if(segTree[i].l==a&&segTree[i].r==b)
    {
        segTree[i].Inc += c;
        return;
    }
    segTree[i].sum+=c*(b-a+1);
    int mid=(segTree[i].l+segTree[i].r)>>1;
    if(b<=mid)  Add(i<<1,a,b,c);  //区间在左边
    else if(a>mid)  Add(i<<1|1,a,b,c); //区间在右边
    else
    {   //区间中间
        Add(i<<1,a,mid,c);
        Add(i<<1|1,mid+1,b,c);
    }
}
LL Query(int i,int a,int b)//查询a-b的总和
{
    if(segTree[i].l==a&&segTree[i].r==b)
    {
        return segTree[i].sum+(b-a+1)*segTree[i].Inc;
    }
    //父级节点加上更新值
    segTree[i].sum+=(segTree[i].r-segTree[i].l+1)*segTree[i].Inc;
    int mid=(segTree[i].l+segTree[i].r)>>1;
    //向下加上更新值
    Add(i<<1,segTree[i].l,mid,segTree[i].Inc);
    Add(i<<1|1,mid+1,segTree[i].r,segTree[i].Inc);
    segTree[i].Inc=0;
    //判断区间并返回
    if(b<=mid)  return Query(i<<1,a,b);
    else if(a>mid)  return Query(i<<1|1,a,b);
    else return Query(i<<1,a,mid)+Query(i<<1|1,mid+1,b);
}

int main(){
    int n;
    while(~scanf("%d",&n),n){
        Build(1,1,n);
        for(int i=0;i<n;i++){
            int l,r;
            scanf("%d%d",&l,&r);
            Add(1,l,r,1);
        }
    for(int i=1;i<=n;i++){
        printf("%d",Query(1,i,i));
        if(i!=n) printf(" ");
        else printf("\n");
    }
    }
}