poj1436(线段树lazy标记）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=8002;
bool vis[maxn][maxn];

struct node{
int y1,y2,x;
}s[maxn];

bool cmp(node a,node b)
{
return a.x<b.x;
}
int cov[maxn<<3];

void Pushdown(int p)
{
if(cov[p])
{
cov[p<<1]=cov[p<<1|1]=cov[p]; //覆盖的线段颜色
cov[p]=0;
}
}

void Update(int p,int l,int r,int x,int y,int cr) //cr为颜色编号
{
if(x<=l&&y>=r)
{
cov[p]=cr;
return;
}
Pushdown(p);
int mid=(l+r)/2;
if(x<=mid) Update(p<<1,l,mid,x,y,cr);//切记
if(y>mid) Update(p<<1|1,mid+1,r,x,y,cr);

}
void Query(int p,int id,int l,int r,int x,int y)
{

if(x<=l&&y>=r){
if(cov[p])
{
vis[cov[p]][id]=true;
return;
}
}
if(l==r) return; //要加，否则无法调用Pushdown(p);
Pushdown(p);
int mid=(l+r)/2;
if(x<=mid) Query(p<<1,id,l,mid,x,y);
if(y>mid) Query(p<<1|1,id,mid+1,r,x,y);
}

int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&s[i].y1,&s[i].y2,&s[i].x);
s[i].y1*=2;
s[i].y2*=2;
}
sort(s+1,s+1+n,cmp);
int sum=0;
memset(cov,0,sizeof(cov));
memset(vis,false,sizeof(vis));

for(int i=1;i<=n;i++)
{
Query(1,i,0,maxn*2,s[i].y1,s[i].y2);
Update(1,0,maxn*2,s[i].y1,s[i].y2,i);

}
for(int i=1;i<=n;i++)
for(int j=i+1;j<=n;j++)
{
if(vis[i][j])
{
//cout<<i<<' '<<j<<endl;
for(int k=j+1;k<=n;k++)
if(vis[i][k]&&vis[j][k])
sum++;
}
}
printf("%d\n",sum);

}
return 0;
}