(中等) POJ 1436 Horizontally Visible Segments , 线段树+区间更新。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>

#define lson L,M,po*2
#define rson M+1,R,po*2+1

using namespace std;

const int N=8000*2;

int COL[8008*2*4];
bool map1[8008][8008];

void pushDown(int po)
{
    if(COL[po]>=0)
    {
        COL[po*2]=COL[po*2+1]=COL[po];
        COL[po]=-1;
    }
}

void update(int ul,int ur,int ut,int L,int R,int po)
{
    if(ul<=L&&ur>=R)
    {
        COL[po]=ut;

        return;
    }

    pushDown(po);

    int M=(L+R)/2;

    if(ul<=M)
        update(ul,ur,ut,lson);
    if(ur>M)
        update(ul,ur,ut,rson);

    if(COL[po*2]==COL[po*2+1])          //这里算是一个剪枝操作，可以减少递归次数。
        COL[po]=COL[po*2];
    else
        COL[po]=-1;
}

void query(int ql,int qr,int qt,int L,int R,int po)
{
    if(COL[po]!=-1)
    {
        map1[qt][COL[po]]=map1[COL[po]][qt]=1;
        return;
    }

    if(L==R)
        return;

    int M=(L+R)/2;

    if(ql<=M)
        query(ql,qr,qt,lson);
    if(qr>M)
        query(ql,qr,qt,rson);
}

struct state
{
    int y1,y2,x1;
};

state sta[8008];

int cmp(const void*a,const void*b)
{
    return (*(state*)a).x1-(*(state*)b).x1;
}

int main()
{
    int d;
    cin>>d;

    int n;

    while(d--)
    {
        memset(map1,0,sizeof(map1));
        memset(COL,0,sizeof(COL));

        cin>>n;

        for(int i=0;i<n;++i)
            scanf("%d %d %d",&sta[i].y1,&sta[i].y2,&sta[i].x1);

        qsort(sta,n,sizeof(state),cmp);

        for(int i=1;i<=n;++i)
        {
            query(sta[i-1].y1*2,sta[i-1].y2*2,i,0,N,1);
            update(sta[i-1].y1*2,sta[i-1].y2*2,i,0,N,1);
        }

        long long ans=0;

        for(int i=1;i<=n;++i)           //直接暴力求解，居然不超时。。。
            for(int j=i+1;j<=n;++j)
                if(map1[i][j])
                    for(int k=j+1;k<=n;++k)
                        if(map1[i][k]&&map1[j][k])
                            ++ans;

        cout<<ans<<endl;
    }



    return 0;
}