数据结构之线段树 入门HDU1556 color the ball

来源:互联网 发布:阿里云服务器1核1g 编辑:程序博客网 时间:2024/05/22 17:48

Color the ball

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


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
#include<iostream>#define mnum 300000using namespace std;typedef struct node{int data;int l,r;}node;node tree[mnum];int n;int father[mnum/3+2]; //记录单个结点对应的数组标号 void build(int left,int right,int i){tree[i].l=left;tree[i].r=right;tree[i].data=0;if(left==right){father[left]=i;return;}build(left,(right+left)/2,2*i);build(((right+left)/2)+1,right,2*i+1);}void insert(int left,int right,int i,int len,int sum)  //对线段树中各点的data(代表color的次数)进行更新,len为待更新区间的长度,sum为已更新的区间的长度!{                                                //这个更新的实现是递归实现,但是还可以尝试着用循环来实现,这样时间效率更高!int mid=(tree[i].l+tree[i].r)/2;if(sum==len)return;if(left==tree[i].l&&right==tree[i].r){tree[i].data++;sum+=tree[i].r-tree[i].l;return;} if(right<mid+1) //如果区间在中值得左边,找左子树 {insert(left,right,2*i,len,sum);}else if(left>mid)//如果区间在中值的右边,找右子树 {insert(left,right,2*i+1,len,sum);}else //前面两种情况如果都不属于的话,那么待寻找的区间就就部分在了左子树上,且部分在右子树上{insert(left,mid,2*i,len,sum);insert(mid+1,right,2*i+1,len,sum);}} int read(int i){int cnt=0;int t=father[i];while(1){cnt+=tree[t].data;if(t==1)return cnt;t/=2;}}int main(){while(cin>>n&&n){build(1,n,1);int a,b;for(int i=1;i<=n;i++){cin>>a>>b;insert(a,b,1,b-a+1,0);}for(int i=1;i<=n;i++){cout<<read(i);if(i!=n)cout<<" ";else cout<<endl;}}}总结一下,使用线段树后,时间复杂度对每组数据来讲是O(2lgn)
0 0
原创粉丝点击