题目链接: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;
}