poj 1436 Horizontally Visible Segments 线段树求垂直三角形的个数

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

题意：三条垂直线如果能两两用水平线相连，且该水平线不被和其他的垂直线相交，则这三条垂直线成为垂直三角形，然后给你一些线段，数据有y的坐标范围和x的坐标，求所有这些线能够成的垂直三角形的个数

分析：用线段树解决，根据线段x坐标从小到大插入的线段树中，每次插入前先查询这个线段能够看见的线段，把这些线段的id存到这条新要插入的线段的vector中，以便后来计算

线段树节点代表该区间内的线段的id，如果该区间有多条线段，则用-1表示。

每次查找的时候，找到sum值不为-1或者l==r为止，并且，在查当前线段时，避免重复插入它能看到的线段，所以用vis数组来存储某些段是否已经存入

在所有的线段都插入完成之后，得到了一个vector数组，里面存的是每条线段能看到的线段的id，按x从大到小去计数，最终得到三角形的个数

以下是代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int MAX = 8010<<1;
vector<int>v[MAX>>1];
struct line
{
    int x,y1,y2;
}l[MAX];
int sum[MAX<<2];
inline void push_up(int rt)
{
    sum[rt] = (sum[rt<<1]==sum[rt<<1|1])?sum[rt<<1]:-1;
}
inline void push_down(int rt)
{
    if(sum[rt]!=-1)
    {
        sum[rt<<1] = sum[rt<<1|1] = sum[rt];
    }
}
void build(int l,int r,int rt)
{
    if(l==r) return;
    int mid = r+l>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
}
void update(int p,int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
    {
        sum[rt] = p;
        return;
    }
    push_down(rt);
    int mid = r+l>>1;
    if(L<=mid) update(p,L,R,l,mid,rt<<1);
    if(R>mid) update(p,L,R,mid+1,r,rt<<1|1);
    push_up(rt);
}
int vis[MAX>>1];
void query(int x,int L,int R,int l,int r,int rt)
{
    if(sum[rt]!=-1)
    {
        if(vis[sum[rt]]!=x)
        {
            vis[sum[rt]] = x;
            v[x].push_back(sum[rt]);
        }
        return;
    }
    if(l==r) return;
    int mid = r+l>>1;
    if(L<=mid) query(x,L,R,l,mid,rt<<1);
    if(R>mid) query(x,L,R,mid+1,r,rt<<1|1);

}
int cmp(const line &l1,const line &l2)
{
    return l1.x<l2.x;
}
int main()
{
    int cas;
    scanf("%d",&cas);
    build(0,MAX,1);
    while(cas--)
    {
        memset(sum,-1,sizeof(sum));
        int n;
        scanf("%d",&n);
        for(int i=0; i<n; i++)
        {
            scanf("%d%d%d",&l[i].y1,&l[i].y2,&l[i].x);
            l[i].y1 <<= 1;
            l[i].y2 <<= 1;
        }
        sort(l,l+n,cmp);
        memset(vis,0,sizeof(vis));
        for(int i=0; i<n; i++)
        {
            query(n-i,l[i].y1,l[i].y2,0,MAX,1);
            update(n-i,l[i].y1,l[i].y2,0,MAX,1);
        }
        int ans = 0;
        for(int i=1; i<=n; i++)
        {
            int size1 = v[i].size();
            for(int j=0; j<size1-1; j++)
            for(int k=j+1; k<size1; k++)
            {
                int tmp = v[i][j],tmp1 = v[i][k];
                if(tmp>tmp1) {int x0=tmp;tmp=tmp1;tmp1=x0;}
                int size2 = v[tmp].size();
                for(int ll=0; ll<size2; ll++)
                if(v[tmp][ll]==tmp1)
                {
                    ans++;
                    break;
                }
            }
            v[i].clear();
        }
        printf("%d\n",ans);
    }
    return 0;
}