HDU-1556 Color the ball(线段树-区间染色)

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

题目大意&&解题思路:

    给出n个气球,通过n次操作给区间a到b的气球染色,最终输出每个气球被染色的次数,n量级为10的5次方所以用数组强行模拟会TLE。通过线段树操作,tr[i].l代表第i个节点的左区间,tr[i].r代表第i个节点的右区间,cover为标记变量表示一段区间被染色次数(注意,不需要染色至叶节点,只需染能a到b拆分后的几个子区间即可)。


Code:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cstring>
using namespace std;
const int maxn=1000000+5;
int n,p;
int ans;
struct Treenode{
    int l,r;
    int cover;
    int lazy;
}tr[maxn<<2];
void Build_tree(int l,int r,int i){
    tr[i].l=l;
    tr[i].r=r;
    tr[i].cover=0;
    if(l==r) return;
    Build_tree(l, (l+r)>>1, i<<1);
    Build_tree(((l+r)>>1)+1, r, i<<1|1);
}
void pushup(int i){
    tr[i].cover++;
    if(tr[i].l!=tr[i].r){
        pushup(i<<1);
        pushup(i<<1|1);
    }
}

void Insert_tree(int L,int R,int i){
    if(L<=tr[i].l && tr[i].r<=R) {
        tr[i].cover++;
        return;
    }
    if(tr[i].l!=tr[i].r){
        int mid=(tr[i].l+tr[i].r)>>1;
        if(L<=mid) Insert_tree(L, R, i<<1);
        if(R>mid) Insert_tree(L, R, i<<1|1);
    }
}

void Output_tree(int i){
    if(tr[i].l==tr[i].r) {
        if(!p){printf("%d",ans+tr[i].cover);p=1;
        } 
        else printf(" %d",ans+tr[i].cover);
    }
    else{
        ans+=tr[i].cover;
        Output_tree(i<<1);
        Output_tree(i<<1|1);
        ans-=tr[i].cover;
    }
}

int main(){
    while(scanf("%d",&n) && n){
        p=0;
        ans=0;
        Build_tree(1,n,1);
        int a,b;
        for(int i=0;i<n;i++){
            scanf("%d %d",&a,&b);
            Insert_tree(a,b,1);
        }
        Output_tree(1);
        printf("\n");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_39665840/article/details/79890122
今日推荐