ZOJ - 1610 -- Count the Colors【线段树 + 区间更新】

Count the Colors

题意：

将一条线染色，先染色的线段可能会被后染色的覆盖，问染完色后，每种颜色有多少段，某个颜色不在直线上就不输出。

思路；

将线段颜色，处理成点的颜色，某个线段被染色处理为这个区间的所有点被染成一种色，用一个color数组记录每个点的颜色，初始化为-1，也就是未被染色，ans数组记录每种颜色一共有多少段，用prev表示前一个点的颜色，查询时记录每个颜色有几段。

AC代码

/*
区间更新 + 暴力求解
时间复杂度O(t * min(maxn, n * log(maxn)))
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define hhh 	puts("hhh");
#define ls 		(rt << 1), l, m 
#define rs 		(rt << 1 | 1), (m + 1), r
const int maxn = 8005;
int color[maxn << 2], ans[maxn], prev;
void pushDown(int rt){
	color[rt << 1] = color[rt];
	color[rt << 1 | 1] = color[rt];
	color[rt] = -1;
}
void update(int rt, int l, int r, int L, int R, int C){
	if (L <= l && r <= R){
		color[rt] = C;
		return;
	}
	if (color[rt] != -1){
		pushDown(rt);
	}	
	int m = (l + r) >> 1;
	if (L <= m)		update(ls, L, R, C);
	if (R > m)		update(rs, L, R, C);
}
void query(int rt, int l, int r){
	if (l == r){
		if (color[rt] != -1 && prev != color[rt]){
			ans[color[rt]]++;
		}
		prev = color[rt];
		return;
	}
	if (color[rt] != -1){
		pushDown(rt);
	}
	int m = (l + r) >> 1;
	query(ls);
	query(rs);
}
void solve(){
	int n, x1, x2, c;
	while (scanf("%d", &n) != EOF){
		memset(ans, 0, sizeof(ans));
		memset(color, -1, sizeof(color));
		while (n--){
			scanf("%d%d%d", &x1, &x2, &c);
			/*由于题目给出的是染线段颜色，不会给出点， 考虑到左端点为0的情况，
			将所有区间都往右移动一为，不会影响结果。 
			*/ 
			update(1, 1, maxn, x1 + 1, x2, c);
		}
		prev = -1;
		query(1, 1, maxn);
		for (int i = 0; i < maxn; ++i){
			if (ans[i] != 0){
				printf("%d %d\n", i, ans[i]);
			}
		}
		puts("");
	}
}
int main(){
	solve();
	return 0;
}