【gmoj】【NOIP2013模拟联考11】 【树状数组】 【离散化】矩形

【gmoj】【NOIP2013模拟联考11】 【树状数组】 【离散化】矩形

题目

---

解题思路

先将点的坐标离散化
然后按高低排序

• 先枚举较低的一条横边 选出满足条件的竖边
• 再枚举高的横边 剔除掉不符合条件的

满足条件的竖边个数用树状数组维护
最后得到满足条件的竖边个数为tot
那么ta对答案的贡献为 tot（tot-1）/2

---

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
struct lzf{
	int x1,y1,x2,y2,p;
}f[2020],l[2020],r[2020];
struct yty{
	int x,id,c;
}h[10010];
int n,t,tl,tr,x1,x2,y1,y2,ans,tot,tree[1000200],zc[2020];
bool cmp(yty t,yty f)
{
	 return t.x<f.x;
}
bool cmp2(lzf t,lzf f)
{
	 if (t.y1!=f.y1)
	    return t.y2<f.y2;
	     else return t.y1<f.y1; 
}
void add(int x,int y)
{
	 for (;x<=h[t].x;x+=x&(-x))
	     tree[x]+=y;
}
int ask(int x)
{
	int sum=0;
	for (;x;x-=x&(-x))
	    sum+=tree[x];
	return sum;
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
		 scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
		 if (x1==x2)  //判断是竖边还是横边
		 {
		    if (y1>y2) swap(y1,y2);
		     f[i]=(lzf){x1,y1,x2,y2,2};	
		 }
		 else {
		 	if (x1>x2) swap(x1,x2);
		 	f[i]=(lzf){x1,y1,x2,y2,1};
		 }
		 h[++t]=(yty){x1,i,1};
		 h[++t]=(yty){y1,i,2};
		 h[++t]=(yty){x2,i,3};
		 h[++t]=(yty){y2,i,4};  
	}
	sort(h+1,h+t+1,cmp);
	int w=h[1].x;
	h[1].x=1;
	for (int i=2;i<=t;i++)  //离散化
	    if (h[i].x==w) h[i].x=h[i-1].x;
	       else w=h[i].x,h[i].x=h[i-1].x+1;
	for (int i=1;i<=t;i++)  
	{
		int k=h[i].id;
		if (h[i].c==1) f[k].x1=h[i].x;
		   else if (h[i].c==2) f[k].y1=h[i].x;
		           else if (h[i].c==3) f[k].x2=h[i].x;
				        else f[k].y2=h[i].x;
	}
	for (int i=1;i<=n;i++)   //分类，横or竖
	    if (f[i].p==2)
	        l[++tl]=f[i];
	        else r[++tr]=f[i];
	sort(l+1,l+tl+1,cmp2);
	sort(r+1,r+tr+1,cmp2);   //按从低到高排序
	for (int i=1;i<tr;i++)  //枚举较低的横边
	{
	    memset(tree,0,sizeof(tree));
		lzf x=r[i];
		int head=1,tail=0;
		for (int j=1;j<=tl;j++) 
		    if (l[j].y1<=x.y1&&l[j].y2>=x.y2)
		       if (l[j].x1>=x.x1&&l[j].x2<=x.x2)  //筛去符合条件的
			       add(l[j].x1,1),zc[++tail]=j;
		for (int j=i+1;j<=tr;j++)
		{ 
			lzf y=r[j]; 
			while (head<=tail&&l[zc[head]].y2<y.y1)  //踢掉不符合条件的，第二条横边后面会越来越高，如果当前就比它小，那后面也一定不满足
			{
				  add(l[zc[head]].x1,-1);
				  head++;
			}
			if (head>=tail) break;
			int L=max(x.x1,y.x1),R=min(x.x2,y.x2);
			tot=ask(R)-ask(L-1);   //查询个数
			ans+=tot*(tot-1)/2;   //统计答案
		}
	}	 
	printf("%d",ans);
	return 0;   
}