poj 1436 线段树

题意：给你N条线段（垂直于x轴）的两个y坐标还有x坐标，问相互看到的三元组有多少个。
有点纠结就是，如果两个连线之间正好有一条线段的某个端点，这个也是不能计算的，所以这个端点就有意义了，所以就用上面那个题的做法，全部扩大二倍再用线段树。
Sample Input
1       //测试次数
5       //线段数目
0 4 4   //y1,y2,x
0 3 1
3 4 2
0 2 2
0 2 3
Sample Output

1

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
using namespace std;
#define MOD 1000000007
const int INF=0x3f3f3f3f;
const double eps=1e-5;
#define cl(a) memset(a,0,sizeof(a))
#define ts printf("*****
");
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root 1,n,1
#define mid ((l+r)>>1)
const int MAXN=20000;
int n,m,t,Min,tt;
int sum[MAXN<<2],col[MAXN<<2],hash[MAXN];

vector<int> v[MAXN];
struct node
{
    int s,t,x;
    void in()
    {
        scanf("%d%d%d",&s,&t,&x);
        s<<=1,t<<=1;
    }
}a[MAXN];
bool cmp(node A,node B)
{
    return A.x<B.x;
}
void pushdown(int rt)
{
    if(col[rt]!=-1)
    {
        col[rt<<1]=col[rt<<1|1]=col[rt];
        col[rt]=-1;
    }
}
void update(int L,int R,int val,int l,int r,int rt)
{
    if(l>=L&&r<=R)
    {
        col[rt]=val;
        return;
    }
    pushdown(rt);
    if(L<=mid) update(L,R,val,lson);
    if(R>mid) update(L,R,val,rson);
}
void query(int L,int R,int val,int l,int r,int rt){
    if(col[rt]!=-1){
        if(hash[col[rt]]!=val){ //防止重复覆盖
            v[col[rt]].push_back(val);
            hash[col[rt]]=val;
        }
        return ;
    }
    if(l==r) return ;
    pushdown(rt);
    if(L<=mid) query(L,R,val,lson);
    if(R>mid)  query(L,R,val,rson);
}
int main()
{
    int i,j,k;
    #ifndef ONLINE_JUDGE
    freopen("1.in","r",stdin);
    #endif
    scanf("%d",&tt);
    while(tt--)
    {
        memset(col,-1,sizeof(col));
        memset(hash,-1,sizeof(hash));
        scanf("%d",&n);
        for(i=0;i<n;i++)
        {
            a[i].in();
            v[i].clear();
        }
        sort(a,a+n,cmp);
        for(i=0;i<n;i++)
        {
            query(a[i].s,a[i].t,i,0,16000,1);
            update(a[i].s,a[i].t,i,0,16000,1);
        }
        for(i=0;i<n;i++)
        {
            sort(v[i].begin(), v[i].end());         //v[i]储存的是i号结点所能看到的线段编号
        }
        int tot=0;
        for(i=0;i<n;i++)
        {
            int len=v[i].size();
            for(j=0;j<len;j++)
            {
                for(k=j+1;k<len;k++)
                {
                    int a1=v[i][j];
                    int a2=v[i][k];
                    if(binary_search(v[a1].begin(),v[a1].end(),a2)) tot++;
                }
            }
        }
        printf("%d
",tot);
    }
}