Sample Input
3 //三组数 1 1 //涂气球的区间 2 2 3 3 3 1 1 1 2 1 3 0
Sample Output
1 1 1 //第一个气球被涂1次颜色 第二个气球被涂2次颜色 第三个气球被涂3次颜色 3 2 1
#include <iostream>
#include <stdio.h>
using namespace std;
#define MAXN 100010
struct Node{
int L,R;
int val; //被涂过的次数
};
Node a[MAXN*3+1];
void Init(int d,int l,int r) //初始化线段树
{
if(l==r){ //递归出口
a[d].L = l;
a[d].R = r;
a[d].val = 0;
return ;
}
//初始化当前节点
a[d].L = l;
a[d].R = r;
a[d].val = 0;
//递归初始化孩子节点
int mid = (l+r)/2;
Init(d*2,l,mid);
Init(d*2+1,mid+1,r);
}
void Update(int d,int l,int r) //更新某一区间的值
{
if(a[d].L==l && a[d].R==r){ //递归出口。找到区间
a[d].val++;
return ;
}
if(a[d].L==a[d].R) //递归出口。没有找到
return ;
//没找到
int mid = (a[d].L+a[d].R)/2;
if(mid>=r){ //去左孩子找
Update(d*2,l,r);
}
else if(mid<l){ //去右孩子找
Update(d*2+1,l,r);
}
else { //中点在要查询区间的中间,两边都要找
Update(d*2,l,mid);
Update(d*2+1,mid+1,r);
}
}
int Query(int d,int l,int r) //查询
{
if(a[d].L==l && a[d].R==r) //找到区间
return a[d].val;
if(a[d].L==a[d].R)
return 0;
int mid = (a[d].L+a[d].R)/2;
if(mid>=r){ //去左孩子找
return a[d].val + Query(d*2,l,r);
}
else if(mid<l){ //去右孩子找
return a[d].val + Query(d*2+1,l,r);
}
else { //中点在要查询区间的中间,两边都要找
return a[d].val + Query(d*2,l,mid) + Query(d*2+1,mid+1,r);
}
}
int main()
{
int N,A,B,i,sum;
while(scanf("%d",&N)!=EOF && N){
Init(1,1,N);
for(i=1;i<=N;i++){ //输入并更新线段树
scanf("%d%d",&A,&B);
Update(1,A,B);
}
for(i=1;i<=N;i++){ //输出每一个气球被涂过的次数
sum = Query(1,i,i);
printf("%d",sum);
if(i!=N) printf(" ");
}
printf("\n");
}
return 0;
}
转自 点击打开链接