spoj 2211. Mars Map

线段树求矩形面积和，模板题（数据全部为整数）

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define MAX 10010
#define LCH(n) ((n)<<1)
#define RCH(n) ((n)<<1|1)

struct segment
{
   int l,r,h,f;
}ss[2*MAX];
struct node
{
   int l,r,cnt,len;
   int mid()
   { return (l+r)>>1; }
}tt[2*MAX*4];
int pos[2*MAX];
int nums;

int cmp(struct segment p ,struct segment q)
{
   return p.h<q.h;
}

void build(int a ,int b ,int rt)
{
   tt[rt].l=a; tt[rt].r=b; tt[rt].cnt=0; tt[rt].len=0;
   if(tt[rt].l==tt[rt].r) return ;
   int mid=tt[rt].mid();
   build(a,mid,LCH(rt));
   build(mid+1,b,RCH(rt));
}

int binary(int key ,int low ,int high)
{
   while(low<=high)
   {
      int mid=(low+high)>>1;
      if(key==pos[mid]) return mid;
      else if(key<pos[mid]) high=mid-1;
      else                  low=mid+1;
   }
   return -1;
}

void cal(int rt)
{
   if(tt[rt].cnt)
      tt[rt].len= pos[tt[rt].r+1]-pos[tt[rt].l];
   else if(tt[rt].l==tt[rt].r)
      tt[rt].len=0;
   else
      tt[rt].len=tt[LCH(rt)].len + tt[RCH(rt)].len;
}

void updata(int a, int b ,int val ,int rt)
{
   if(tt[rt].l==a && tt[rt].r==b) //目标区间
   {
      tt[rt].cnt += val;
      cal(rt);
      return ;
   }
   int mid=tt[rt].mid();
   if(b<=mid)
      updata(a,b,val,LCH(rt));
   else if(a>mid)
      updata(a,b,val,RCH(rt));
   else
   {
      updata(a,mid,val,LCH(rt));
      updata(mid+1,b,val,RCH(rt));
   }
   cal(rt);
}

int main()
{
   int n;
   while(scanf("%d",&n)!=EOF)
   {
      nums=0;
      for(int i=0; i<n; i++)
      {
         int x1,y1,x2,y2;
         scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
         ss[nums].l=x1; ss[nums].r=x2; ss[nums].h=y1; ss[nums].f=1;
         ss[nums+1].l=x1; ss[nums+1].r=x2; ss[nums+1].h=y2; ss[nums+1].f=-1;
         pos[nums]=x1; pos[nums+1]=x2;
         nums+=2;
      }
      sort(ss,ss+nums,cmp);
      sort(pos,pos+nums);
      int m=1;
      for(int i=1; i<nums; i++)
         if(pos[i]!=pos[i-1])
            pos[m++]=pos[i];

      build(0,m-1,1);
      long long ans=0;
      for(int i=0; i<nums-1; i++)
      {
         int l=binary(ss[i].l,0,m-1);
         int r=binary(ss[i].r,0,m-1)-1;
         updata(l,r,ss[i].f,1);
         ans += tt[1].len*(ss[i+1].h-ss[i].h);
         //printf("%lld\n",ans);
      }
      printf("%lld\n",ans);
   }
}