hdu 1264 Counting Squares（线段树 + 面积的并）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define mem(a,x) memset(a,x,sizeof(a))
#define ls i<<1
#define rs i<<1|1
#define m(i) ((q[i].l + q[i].r)>>1)
using namespace std;
typedef long long ll;
const int N = 40000;
long long x[2*N];//横坐标
struct Edge
{
    int l,r;//这条线的左右端点的横坐标
    int h;//这条线的纵坐标
    int f;//这条线是矩形的上边还是下边
}e[N<<1];
bool cmp(Edge a,Edge b)
{
    return a.h < b.h;
}
struct Node
{
    int l,r;//横坐标的区间，是横坐标数组的下标
    int s;//该节点被覆盖的情况（是否完全覆盖）
    int len;//该区间被覆盖的总长度
}q[N*8];
void build(int i,int l,int r)
{
    q[i].l = l,q[i].r = r;
    q[i].s = 0;q[i].len = 0;
    if (l == r) return;
    int mid = m(i);
    build(ls,l,mid);
    build(rs,mid+1,r);
}
void pushup(int i)
{
    if (q[i].s) //非零，已经被整段覆盖
    {
        q[i].len = x[q[i].r+1] - x[q[i].l];
    }
    else if (q[i].l == q[i].r) //这是一个点而不是线段
    {
        q[i].len = 0;
    }
    else //是一条没有整个区间被覆盖的线段，合并左右子的信息
    {
        q[i].len = q[ls].len + q[rs].len;
    }
}
void update(int i,int l,int r,int xx)//这里深刻体会为什么令下边为1，上边-1
{                                   //下边插入边，上边删除边
    if (q[i].l == l&&q[i].r == r)
    {
        q[i].s += xx;
        pushup(i);//更新区间被覆盖de总长度
        return;
    }
    int mid = m(i);
    if (r <= mid) update(ls,l,r,xx);
    else if (l > mid) update(rs,l,r,xx);
    else
    {
        update(ls,l,mid,xx);
        update(rs,mid+1,r,xx);
    }
    pushup(i);
}
int main()
{
	while(1){
		int x1,x2,y1,y2;
	    int tot = 0,flag=0;
	    while (scanf("%d %d %d %d",&x1,&y1,&x2,&y2)!=EOF)
	    {
	    	if(x1==-2){
	    	flag=1;
			break;	
			}
	      	if(x1==-1)break;

			if(x1>x2)swap(x1,x2);
			if(y1>y2)swap(y1,y2);  	
	        Edge &t1 = e[tot];Edge &t2 = e[1+tot];
	        t1.l = t2.l = x1,t1.r = t2.r = x2;
	        t1.h = y1;t1.f = 1;
	        t2.h = y2;t2.f = -1;
	        x[tot] = x1;x[tot+1] = x2;
	        tot += 2;  
		}
		
			sort(e,e+tot,cmp);//边按高度从小到大排序（自下而上扫描）
    		sort(x,x+tot);
		        //离散化横坐标
	        int k = 1;
	        for (int i = 1;i < tot;++i)
	        {
	            if (x[i] != x[i-1]) //去重
	            {
	                x[k++] = x[i];
	            }
	        }
	        build(1,0,k-1);//离散化后的区间是[0，k-1]
	        long long ans = 0;
	        for (int i = 0;i < tot;++i)
	        {
	            //因为线段树维护的是横坐标们的下标，所以对每条边求出其两个横坐标对应的下标
	            int l = lower_bound(x,x+k,e[i].l) - x;//在横坐标数组里找到这条边的位置
	            int r = lower_bound(x,x+k,e[i].r) - x - 1;
	            update(1,l,r,e[i].f);//每扫到一条边就更新横向的覆盖len
	            ans += (e[i+1].h - e[i].h)*q[1].len;//q[1]是整个区间,q[1].k=len是整个区间的有效长度
	            //计算面积就是用区间横向的有效长度乘以两条边的高度差（面积是两条边里面的部分）
	        }			      
	        printf("%lld\n",ans);
			
			if(flag)
			break;  		
    }
    return 0;
}