poj1436(线段树成段更新)

题目链接：poj1436

/*poj1436成段更新（难点：端点的处理）
题意：
    n条竖直线段，每行输入y1,y2,x,分别表示线段下端点，上端点，横坐标
定义两天线段可见：如果一条水平线段连接它们，且在它们之间不经过其它线段。
最后求有多少组3条线段是两两可见的。

思路：首先将所有线段按x坐标的大小进行排序，然后对y轴建树
对每一条线段执行查询操作，查询当前线段区间内可见线段有哪些，并记录，然后更新
    题目给的是线段的端点，不能直接用区间建树，样例0 3 1，3 4 2，两线段端点在一条水平线上
区间建树无法判断此类情况
    将纵坐标全部乘以2，就可以处理端点的问题
0,4,1 和 0,2,2 和 3,4,2这三条线段覆盖的结果是区间0~4通过线段树查找可见线段是两条,
其实是3条(2~3可见另一条)
*/
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
#define lson rt<<1
#define rson rt<<1|1
const int N = 16010;
bool v[8005][8005];//v[i][j]表示第i个线段和第j个线段是可见的
struct seg
{
    int x,y1,y2;
    seg(){}
    seg(int xx, int yy1, int yy2):x(xx),y1(yy1),y2(yy2){}
    bool operator < (const seg &cmp) const{
        return x < cmp.x;
    }
}ss[N>>1];
struct node
{
    int l,r,c;
}s[N<<2];
void build(int l, int r, int rt)
{
    s[rt].l = l;  s[rt].r = r;
    s[rt].c = -1;
    if(l == r) return;
    int mid = (l+r) >> 1;
    build(l, mid, lson);
    build(mid+1, r, rson);
}
void pushdown(int rt)
{
    if(s[rt].c != -1){
        s[lson].c = s[rson].c = s[rt].c;
        s[rt].c = -1;
    }
}
void updata(int c, int l, int r, int rt)
{
    if(l <= s[rt].l && s[rt].r <= r){
        s[rt].c = c;
        return;
    }
    pushdown(rt);
    int mid = (s[rt].l + s[rt].r) >> 1;
    if(l <= mid) updata(c, l, r, lson);
    if(mid < r) updata(c, l ,r, rson);
}
void query(int id, int l, int r, int rt)
{
    if(s[rt].c != -1){
        v[id][s[rt].c] = true;
        return;
    }
    if(s[rt].l == s[rt].r) return;
    int mid = (s[rt].l + s[rt].r) >> 1;
    if(l <= mid) query(id, l, r, lson);
    if(mid < r) query(id, l, r, rson);
}
int main()
{
    int T,n,i,j,k;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        build(0, N, 1);
        memset(v, false, sizeof(v));
        for(i = 0; i < n; i ++){
            int x,y1,y2;
            scanf("%d%d%d",&y1,&y2,&x);
            ss[i] = seg(x, y1<<1, y2<<1);
        }
        sort(ss, ss+n);
        for(i = 0; i < n;i ++){
            query(i, ss[i].y1, ss[i].y2, 1);
            updata(i, ss[i].y1, ss[i].y2, 1);
        }
        int ans = 0;
        for(i = 0; i < n; i ++){//暴力枚举
            for(j = 0; j < n; j ++){
                if(v[i][j])
                for(k = 0; k < n; k ++)
                    if(v[i][k] && v[j][k])
                        ans ++;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}