HDU 1556

Color the ball

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

思路

线段树，数据比较大，直接暴力循环肯定会TLE，可以把一开始没被涂过的气球看作是一棵标记全部为0的线段树，然后，录入多次涂色的时候，尽可能找到符合的靠上的区域（假设现在有一棵范围是[1,10]的树，输入[1,6]，我需要找到 [1,5] [6,6] 的范围，然后该区域的标记+=1），然后输出的时候从上往下，按顺序在找到对应的 [I,I] 之前，（假设需要还是那棵10的树，需要找1的涂色次数，那么就从上往下找范围直到[1,1]，分别是[1,10]→[1,5]→[1,3]→[1,2]→[1,1]，把每一个范围的标记加起来就是最终结果）。

代码

#include<stdio.h>
#include<string.h>

int a[400000+5];
int b[100000+5];

void update_tree(int a[],int node,int start,int end,int L,int R) {
	if(start >= L && end <= R) {
		a[node]++;
	}
	else if(L > end || R < start) {
		return;
	}
	else {
		int mid = (start + end) / 2;
		int left_node  = 2 * node + 1;
		int right_node = 2 * node + 2;
		update_tree(a,left_node,start,mid,L,R);
		update_tree(a,right_node,mid+1,end,L,R);
	}
}

void sum_tree(int a[],int b[],int node,int start,int end,int idx) {
	if(idx >= start && idx <= end) {
		b[idx] += a[node];
	}
	if(start == end) {
		return;
	}
	if(idx < start || idx > end) {
		return;
	}
	int mid = (start + end) / 2;
	int left_node  = 2 * node + 1;
	int right_node = 2 * node + 2;
	sum_tree(a,b,left_node,start,mid,idx);
	sum_tree(a,b,right_node,mid+1,end,idx);
}

int main() {
	int n;
	while(scanf("%d",&n) != EOF) {
		if(n == 0) {
			return 0;
		}
		int L,R,start = 1,end = n;
		memset(a,0,sizeof(a));
		memset(b,0,sizeof(b));
		for(int i=1;i<=n;i++) {
			scanf("%d %d",&L,&R);
			update_tree(a,0,start,end,L,R);
		}
		for(int i=1;i<=n;i++) {
			sum_tree(a,b,0,start,end,i);
			if(i == n) {
				printf("%d\n",b[i]);
			}
			else {
				printf("%d ",b[i]);
			}
		}
	}
	return 0;
}

题目来源

HDU 1556 Color the ball