hdu1556 Color the ball 线段树

Color the ball

Time Limit: 9000/3000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 17146 Accepted Submission(s): 8570



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

3
1 1
2 2
3 3
3
1 1
1 2
1 3
0

Sample Output

1 1 1
3 2 1

Author

8600

Source

HDU 2006-12 Programming Contest

这道题的代码就直接用军营数人数的那个改改就过了，输出会查最后一个空格

因为这个人是对区间内的气球进行涂色，所以每次输入区间的时候就对这个区间做＋＋

然后在线段树中查询单个气球的时候从上至下地累积每个气球所在区间的值，因为这个区间被涂过色的话这个气球必定也被涂过色

输出。。

下面代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=100000;
struct unit{
int l;
int r;
int sum;
}save[(maxn+5)*4];
int a[maxn+5];
void build(int l,int r,int index){
save[index].l=l;
save[index].r=r;
if(l==r){
return;
}
int middle = (l+r)/2;
build(l, middle, 2*index);
build(middle+1, r, 2*index+1);
}

void update(int l,int r,int index){
if(save[index].l==l&&save[index].r==r){
save[index].sum++;
return;
}
int middle = (save[index].l+save[index].r)/2;
if(middle<l){
update(l, r, 2*index+1);
}else if(middle>=r){
update(l, r, 2*index);
}else{
update(l, middle, 2*index);
update(middle+1, r, 2*index+1);
}
}
int getSum(int l,int r,int &sum,int index){
sum += save[index].sum;
if(save[index].l==l&&save[index].r==r){
return sum;
}
int middle = (save[index].l+save[index].r)/2;
if(middle<l){
getSum(l, r, sum, 2*index+1);
}else if(middle>=r){
getSum(l,r, sum, 2*index);
}else{
getSum(l, middle, sum,2*index);
getSum(middle+1, r, sum, 2*index+1);
}
return sum;
}
int main(){
int n,i;
while(scanf("%d",&n)&&n){
memset(save, 0, sizeof(save));
int a,b;
build(1, n, 1);
for(i=1;i<=n;i++){
scanf("%d%d",&a,&b);
update(a, b, 1);
}
for(i=1;i<=n;i++){
int sum=0;
getSum(i, i, sum, 1);
printf("%d",sum);
if(i<n){
printf(" ");
}
}
printf("\n");
}

return 0;
}