T^TOJ长方形的并的面积 线段树+平移扫描法

T^TOJ 长方形的并的面积

线段树维护[ 1 ,  n ]全区间大于0的节点的个数即为矩形的长度,线段之间的宽度即为矩形的宽度

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int MAX=1e3+5;
struct node
{
    int l,r,y,v;
}edge[MAX];
int hash[MAX],sum[MAX],mark[MAX],c1[MAX],c2[MAX],n=3;
bool cmp(node a,node b)
{
    return a.y<b.y;
}
int find(int x)
{
    int l=1,r=n*2,mid=0;
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(hash[mid]<x) l=mid+1;
        else if(hash[mid]>x) r=mid-1;
        else break;
    }
    return mid;
}
void pushup(int rt,int l,int r)
{
	if(mark[rt]) sum[rt]=hash[r+1]-hash[l];
	else if(l==r) sum[rt]=0;
	else sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void update(int rt,int l,int r,int L,int R,int v)
{
	if(L<=l&&R>=r)
	{
	    mark[rt]+=v;
	    pushup(rt,l,r);
	    return;
    }
    int mid=(l+r)>>1;
    if(L<=mid) update(rt<<1,l,mid,L,R,v);
    if(R>mid) update(rt<<1|1,mid+1,r,L,R,v);
    pushup(rt,l,r);
}
int main()
{
    int t,a,b,c,d;
    scanf("%d",&t);
    while(t--)
    {
        int ans=0;
        memset(sum,0,sizeof(sum));
        memset(mark,0,sizeof(mark));
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d%d%d",&a,&b,&c,&d);
            edge[i*2-1].l=edge[i*2].l=a;
            edge[i*2-1].r=edge[i*2].r=b;
            edge[i*2-1].y=c;edge[i*2].y=d;
            edge[i*2-1].v=1;edge[i*2].v=-1;
            hash[i*2-1]=a;hash[i*2]=b;
        }
        sort(hash+1,hash+n*2+1);
        sort(edge+1,edge+n*2+1,cmp);
        for(int i=1;i<n*2;i++)
        {
            int L=find(edge[i].l),R=find(edge[i].r)-1;
            if(L<=R) update(1,1,2*n,L,R,edge[i].v);
            ans+=sum[1]*(edge[i+1].y-edge[i].y);
        }
        printf("%d\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Nrtostp/article/details/81486615