CF EDU 68 E树状数组判断直线之间构成矩形的个数

#include<bits/stdc++.h>
using namespace std;
const int maxn=200000;
long long tr[maxn+100];
struct node{
    int x,l,r;
};
node q[200000],p[200000];
int n;
void add(int x,long long t)
{
    while(x<=maxn)
    {
        tr[x]+=t;
        x+=x&(-x);
    }
}
long long query(int x)
{
    long long ans=0;
    while(x>0)
    {
        ans+=tr[x];
        x-=x&(-x);
    }
    return ans;
}
bool cmp(node a,node b)
{
    return a.x<b.x;
}
struct nod{
    int x,r;
}t[maxn];
bool cmp1(nod a,nod b)
{
    return a.r<b.r;
}
int main(){
	scanf("%d",&n);
	int x1,x2,y1,y2,cp=0,cq=0;
	long long ans=0;
	for(int i=1;i<=n;i++)
	{
	    scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
	    x1+=5001;//保证树状数组中不出现非正数
	    x2+=5001;
	    y2+=5001;
	    y1+=5001;
	    if(x1==x2)
            p[++cp].x=x1,p[cp].l=min(y1,y2),p[cp].r=max(y1,y2);//列
       if(y1==y2)
            q[++cq].x=y1,q[cq].l=min(x1,x2),q[cq].r=max(x1,x2);//行
	}
	sort(p+1,p+1+cp,cmp);
	sort(q+1,q+1+cq,cmp);
	for(int i=1;i<=cp;i++)
    {
        int top=0;
        for(int j=1;j<=cq;j++)
        {
            if(q[j].l<=p[i].x&&q[j].r>=p[i].x&&q[j].x>=p[i].l&&q[j].x<=p[i].r)
            {
                t[++top]=(nod){q[j].x,q[j].r},add(q[j].x,1);//与这一列相交的行存储
            }
        }
        sort(t+1,t+1+top,cmp1);
        int nw=0;
        for(int j=i+1;j<=cp;j++)
        {
            if(p[j].r<p[i].l||p[j].l>p[i].r)continue;
            while(nw<top&&t[nw+1].r<p[j].x)//如果一行的最右点在这一列的左边说明，这一行无法再产生贡献,从树状数组中剖出
            {
                nw++;
                add(t[nw].x,-1);
            }
            long long tt=query(p[j].r)-query(p[j].l-1);
           // cout<<i<<" "<<j<<" "<<tt<<endl;
            ans+=(long long)tt*(tt-1)/2;
        }
        while(nw<top)
        {
            nw++;
            add(t[nw].x,-1);
        }
    }
    printf("%lld\n",ans);
	return 0;
}