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,输入结束。
当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#include<iostream> #define mnum 300000 using 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)