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

31 12 23 331 11 21 30

 

Sample Output

1 1 13 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 longusing 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");    }    }}