hdu1556(线段树)

来源:互联网 发布:巴以冲突 知乎 编辑:程序博客网 时间:2024/05/20 09:44



题意:


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


线段树:

这道题计数的时候不需要一定记到叶子节点(就靠这个来省时间)

最后返值得时候

tree[n*2].value+=tree[n].value;

tree[n*2+1].value+=tree[n].value;

就可以了


//

//  main.cpp

//  BALl

//

//  Created by Mr.Xue on 17/7/19.

//  Copyright © 2017 Mr.Xue. All rights reserved.

//


#include <iostream>

#include <stdio.h>

#include <string.h>

using namespacestd;

int ans[100005],m;

struct node

{

    int l,r,value;

}tree[500000];

void buildTree(int l,int r,int n)

{

    tree[n].l=l;

    tree[n].r=r;

    tree[n].value=0;

    if(l==r)

        return;

    int mid=(l+r)/2;

    buildTree(l,mid,n*2);

    buildTree(mid+1,r,n*2+1);

}

void qurry(int l,int r,int n)

{

    if(l==tree[n].l&&r==tree[n].r)

    {

        tree[n].value++;

        return;

    }

    int mid=(tree[n].l+tree[n].r)/2;

    if(mid>=r)

        qurry(l,r,n*2);

    else if(mid<l)

        qurry(l,r,n*2+1);

    else

    {

        qurry(l,mid,n*2);

        qurry(mid+1,r,n*2+1);

    }

}

void get(int n)

{

    if(tree[n].l==tree[n].r)

    {

        ans[++m]=tree[n].value;

        return;

    }

    tree[n*2].value+=tree[n].value;

    tree[n*2+1].value+=tree[n].value;

    get(n*2);

    get(n*2+1);

}

int main()

{

    int n,a,b;

    while(scanf("%d",&n)&&n!=0)

    {

        m=0;

        buildTree(1,n,1);

        for(int i=1;i<=n;i++)

        {

            scanf("%d %d",&a,&b);

            qurry(a,b,1);

        }

        get(1);

        for(int i=1;i<n;i++)

            printf("%d ",ans[i]);

        printf("%d\n",ans[n]);

    }

    return0;

}


直接算会超时
原创粉丝点击