poj 1436 Horizontally Visible Segments(线段树)

poj 1436 Horizontally Visible Segments

太暴力了、、

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 8005
#define lch p<<1
#define rch p<<1|1
#define mid (t[p].l+t[p].r)>>1
struct node
{
int l,r;
int c,flag;
}t[MAXN<<3];
struct segment
{
int x,yd,yu;
bool operator < (const segment &a) const
{
return x<a.x;
}
}s[MAXN];
bool map[MAXN][MAXN];
void construct(int l,int r,int p)
{
t[p].l=l,t[p].r=r,t[p].c=-1,t[p].flag=0;
if(l==r) return ;
int m=mid;
construct(l,m,lch);
construct(m+1,r,rch);
}
void pushdown(int p)
{
t[lch].c=t[rch].c=t[p].c;
t[p].c=-1;
}
void modify(int l,int r,int c,int p)
{
if(t[p].l>=l&&t[p].r<=r)
{
t[p].c=c;
return ;
}
if(t[p].l==t[p].r) return ;
if(t[p].c!=-1) pushdown(p);
if(l<=t[lch].r) modify(l,r,c,lch);
if(r>=t[rch].l) modify(l,r,c,rch);
}
void query(int l,int r,int c,int p)
{
if(t[p].c!=-1)
{
map[t[p].c][c]=true;
return ;
}
if(t[p].l==t[p].r) return ;
if(t[p].c!=-1) pushdown(p);
if(l<=t[lch].r) query(l,r,c,lch);
if(r>=t[rch].l) query(l,r,c,rch);
}
int main()
{
int cas,n;
scanf("%d",&cas);
while(cas--)
{
scanf("%d",&n);
int ydd=1<<30,yuu=0;
for(int i=0;i<n;i++)
{
scanf("%d%d%d",&s[i].yd,&s[i].yu,&s[i].x);
s[i].yd<<=1;
s[i].yu<<=1;
if(s[i].yd<ydd) ydd=s[i].yd;
if(s[i].yu>yuu) yuu=s[i].yu;
}
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
map[i][j]=false;
construct(ydd,yuu,1);
sort(s,s+n);
for(int i=0;i<n;i++)
{
query(s[i].yd,s[i].yu,i,1);
modify(s[i].yd,s[i].yu,i,1);
}
int ans=0;
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;
}