N个气球排成一排,从左到右依次编号为1,2,3....N.每次给定2个整数a b(a <= b),lele便为骑上他的“小飞鸽"牌电动车从气球a开始到气球b依次给每个气球涂一次颜色。但是N次以后lele已经忘记了第I个气球已经涂过几次颜色了,你能帮他算出每个气球被涂过几次颜色吗?
当N = 0,输入结束。
3 1 1 2 2 3 3 3 1 1 1 2 1 3 0
1 1 1 3 2 1
线段树的区间更新,每次找到要更新的区间,记录加1
最后找到每个区间更新arr数组(记录每个气球被刷的次数)
代码:
#include<stdio.h>
#include<string.h>
#define MAX 100005
using namespace std;
struct node{
int val;
int l,r;//此结点表示的区间范围
}tree[MAX*4];
int arr[MAX];
/*
功能:建立线段树
root:当前根结点
[left,right]:建树区间
*/
void build(int root,int left,int right){
tree[root].l=left;
tree[root].r=right;
tree[root].val=0;
if(left!=right){
int mid=(right+left)/2;
build(root*2,left,mid);
build(root*2+1,mid+1,right);
}
return;
}
/*
功能:区间更新
root:当前根结点
[left,right]:更新区间
*/
void update(int root,int left,int right){
if(tree[root].l==left&&tree[root].r==right){//找到要刷的气球区间,更新其被刷的次数+1
tree[root].val++;
return;
}
else{
int mid=(tree[root].l+tree[root].r)/2;
if(right<=mid) update(root*2,left,right);
else if(left>mid) update(root*2+1,left,right);
else{
update(root*2,left,mid);
update(root*2+1,mid+1,right);
}
}
}
/*
功能:将更新的区间结果记录到arr数组
root:当前根结点
*/
void down_mark(int root){//访问了每个结点
int i;
for(i=tree[root].l;i<=tree[root].r;i++){//此根节点表示的区间都被刷过
arr[i]+=tree[root].val;
}
if(tree[root].l==tree[root].r) return;//到达叶结点
down_mark(root*2);
down_mark(root*2+1);
}
int n;
int main(){
while(scanf("%d",&n)!=EOF){
if(n==0) break;
memset(arr,0,sizeof(arr));
memset(tree,0,sizeof(tree));
build(1,1,n);
int i;
for(i=0;i<n;i++){
int a,b;
scanf("%d%d",&a,&b);
if(a>b){
int c=a;
a=b;b=c;
}
update(1,a,b);
}
down_mark(1);
for(i=1;i<=n-1;i++)
printf("%d ",arr[i]);
printf("%d\n",arr[i]);
}
return 0;
}