首页 > 代码库 > 线段树(hdu 1556)

线段树(hdu 1556)

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个气球总共被涂色的次数。

技术分享
抱歉啊 这几天发的全是线段树 (上周培训的这个,,,,下周就换点其他的)
这个题也很简单 只需要用到 mark 缓存 其实这道题 用普通的 区间起点终点更新 也可以 反正每一个节点都要访问的。
还是贴代码吧
#include<iostream>
#include<stdio.h>
using namespace std;
bool flag;
struct node
{
    int l,r,mark;//直接用mark 标记这一段被刷了几次就可以了  自己画一个图很好理解的 
}data[100001*4];
void built(int l,int r,int i)
{
    data[i].l=l; data[i].r=r; data[i].mark=0;
    if(l==r){return;}
    int mid=(l+r)/2;
    built(l,mid,i*2);
    built(mid+1,r,i*2+1);
    return ;
}
void down_mark(const int& i)
{
    data[i*2].mark+=data[i].mark;
    data[i*2+1].mark+=data[i].mark;
    data[i].mark=0;
}
void refresh(int l,int r,int i)
{
    if(data[i].l==l&&data[i].r==r)
    {
        data[i].mark++; return;
    }
    if(data[i].mark)
    down_mark(i);
    int mid=(data[i].l+data[i].r)/2;
    if(l>=mid+1) refresh(l,r,i*2+1);
    else if(r<=mid) refresh(l,r,i*2);
    else{
        refresh(l,mid,i*2); 
        refresh(mid+1,r,i*2+1); 
    }
}
void query(int l,int r,int i)
{
    if(l==r) 
    { 
    if(flag==0) cout<<data[i].mark;//输出格式需要注意下 
    else cout<<" "<<data[i].mark; 
      flag=1;return ;}
    if(data[i].mark)
    down_mark(i);
    int mid=(l+r)/2;
    query(l,mid,i*2);
    query(mid+1,r,i*2+1);
    
 } 
int n,a,b; 
int main()
{
    while(~scanf("%d",&n),n)
    {
        built(1,n,1);
        for(int i=0;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            refresh(a,b,1);
        }
        flag=0;
        query(1,n,1);
        cout<<endl;
    }
    return 0;
}

 

线段树(hdu 1556)