POJ 1436 Horizontally Visible Segments（线段树建图+枚举）

题目连接：http://poj.org/problem?id=1436

题意：给一些线段，每个线段有三个值y1, y2, x代表起点为（x, y1），终点为（x, y2）的线段。当从一个线段可以作水平线到另一个线段并且不穿过其他线段时，就称这两个线段时水平可见的。当三个线段可以两两水平可见，就称为形成一个线段三角。问：在这些线段中有多少个这样的线段三角？

分析：可以把每条线段看做是一个点，如果它和其他线段是水平可见的，就将这两个点相连，由于是无向图，就是你能看到我，我也能看到你，所以需要连接两次（map[a][b] = 1; map[b][a] = 1;）。然后问题就是如何生成存放线段关系的图，可以用线段树来求。假如现在遍历到第i条线段，那么第i条线段所能看到的是前面区间最新的线段，如图：

线段5只能看到1、2、4，不能看到3，所以可以用线段树区间修改的方法来生成这个树。

初始化set数组（存放线段的标号）全为-1，当遍历到某条线段时，对线段的那段区间进行查询，如果查询到的区间有值，说明可以看到线段set[o]，然后即相连，如果查询到的区间一直是-1，就一直深搜，直到到达叶子节点返回。查询结束之后将当前线段插入到线段树中，即把区间更新。因为都是整数，所以某两条线段可见可能是在0.5的时候可见，所以需要预处理一下线段的区间，将区间扩大至2倍即可。

建完图之后暴力枚举每三个点即可。

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 8000;
struct segment{
	int a, b, x;
	bool operator < (const segment& temp) const {
		return x < temp.x; 
	}
}a[maxn+10];
int set[8*maxn+10];
bool map[maxn+10][maxn+10];
void push_down(int o){
	if (set[o] != -1){
		//将其左孩子和右孩子赋为和父亲一样的标号 
		set[o<<1] = set[o];
		set[o<<1|1] = set[o];
		set[o] = -1;//因为该区间已经不是一条线段全部覆盖了，所以标记重新初始化 
	}
}
void query(int o, int L, int R, int i){
	if (set[o] != -1){//区间有值，说明可以看到set[o]，将其连接 
		map[i][set[o]] = 1;
		map[set[o]][i] = 1;
		return;
	}
	if (L == R){//没有值，还是初始状态，到叶子节点返回 
		return;
	}
	int m = (L+R)>>1;
	if (a[i].a <= m) query(o<<1, L, m, i);
	if (a[i].b > m) query(o<<1|1, m+1, R, i);
}
void insert(int o, int L, int R, int i){
	if (a[i].a <= L && a[i].b >= R){
		set[o] = i;//找到区间，将节点的值赋为i，表示这段区间可以看到第i条线段 
		return;
	}
	push_down(o);//节点信息向下更新 
	int m = (L+R)>>1;
	if (a[i].a <= m) insert(o<<1, L, m, i);
	if (a[i].b > m) insert(o<<1|1, m+1, R, i);
}
int main(){
	int T, n;
	scanf("%d", &T);
	while (T--){
		scanf("%d", &n);
		memset(set, -1, sizeof(set));//初始化为-1 
		memset(map, false, sizeof(map));//初始化为不相连 
		int L = (1<<31)-1, R = -1;
		for (int i=0; i<n; i++){
			scanf("%d%d%d", &a[i].a, &a[i].b, &a[i].x);
			a[i].a*=2;//预处理 
			a[i].b*=2;
			L = min(L, a[i].a);
			R = max(R, a[i].b);
		}
		sort(a, a+n);//将线段按照x从小到大排序 
		for (int i=0; i<n; i++){
			query(1, L, R, i);
			insert(1, L, R, i);
		}
		int ans = 0;
		//枚举三个线段，符合条件结果+1 
		for (int i=0; i<n; i++){
			for (int j=i+1; j<n; j++){
				if (map[i][j]){
					for (int k=j+1; k<n; k++){
						if (map[i][k] && map[j][k]){
							ans++;
						}
					}
				}
			}
		}
		printf("%d\n", ans);
	}
	return 0;
}