线段树的应用-poj2777的解法

poj2777的问题描述：

给一个板子染色，板子长L，均分成L段，分别标记为1,2, ..., L，每段只能染一个颜色（颜色种类很少，<=30）。

有两种操作：

（1）C A B C：从段A到段B，染C颜色。

（2）P A B：查询输出段A到段B共有几种不同的颜色。

线段树代码：

#include <stdio.h> #define LEN 100000 #define COLOR 1 typedef struct segment { int l; int r; int color; // bits:[1,30] will be used }segment; segment s[LEN*3]; int color; /* * S_init * 构造线段树 * [l, r] - 节点区间 * index - 存储位置 */ void S_init(int l, int r, int index) { int mid; s[index].l = l; s[index].r = r; s[index].color = COLOR; if (l == r) return; mid = (l + r) >> 1; S_init(l, mid, index*2); // 左孩子 S_init(mid+1, r, index*2+1);// 右孩子 } /* * Count * 变量的二进制表示中，1的个数 */ int Count(int color) { int cnt = 0; while (color) { cnt += (color & 1); color = color >> 1; } return cnt; } void S_color(int l, int r, int index, int a, int b, int d) { int mid = 0; // 染色 if (a == l && b == r) // [l,r] = [a,b] { s[index].color = d; return; } if (s[index].color == d) return; // 单一颜色，向下传递一层 if (Count(s[index].color) == 1) { s[index*2].color = s[index].color; s[index*2+1].color = s[index].color; } mid = (l + r) >> 1; if (mid >= b) S_color(l, mid, index*2, a, b, d); else if (mid < a) S_color(mid+1, r, index*2+1, a, b, d); else { S_color(l, mid, index*2, a, mid, d); S_color(mid+1, r, index*2+1, mid+1, b, d); } // 向上传递一层 s[index].color = s[index*2].color | s[index*2+1].color; } void S_query(int l, int r, int index, int a, int b) { int mid = 0; if ((l==a && r==b) || (l<=a && r>=b && Count(s[index].color)==1)) //区间相等，或，节点区间包含[a, b]且该节点颜色只有一种。 { color = color | s[index].color; return; } mid = (l + r) >> 1; if (mid >= b) S_query(l, mid, index*2, a, b); else if (mid < a) S_query(mid+1, r, index*2+1, a, b); else // 分开 { S_query(l, mid, index*2, a, mid); S_query(mid+1, r, index*2+1, mid+1, b); } } /* * 1，构造线段树 * 2，增加：区间相等，修改节点并与兄弟节点相或(|)的结果存放入父亲节点，直至根节点。 * 3，查询：区间相等，将结果放入全局变量 */ int main() { int L, T, O; int a, b, c, tmp; char q[2]; scanf("%d %d %d", &L, &T, &O); S_init(1, L, 1); while (O--) { scanf("%s %d %d", q, &a, &b); if (a > b) { tmp = a; a = b; b = tmp; } if (q[0] == 'P') // query { color = 0; S_query(1, L, 1, a, b); printf("%d/n", Count(color)); } else // color { scanf("%d", &c); S_color(1, L, 1, a, b, 1 << (c-1)); } } return 0; }

说明：如果某个非叶节点是某一颜色，那么他的所有子孙节点都应该是同一个颜色。

所以，我们在颜色经过该节点时，才向下染色。