hdu 4419 Colourful Rectangle 杭州赛区 1010 （解题报告）

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>

#define ls rt<<1
#define rs rt<<1|1
#define lson l,m,ls
#define rson m+1,r,rs
#define sf scanf
#define pf printf
using namespace std;

const int N = 10000 * 2 +10;

//437MS 2512K 3151 B C++
/*
比赛的时候也想到了容斥原理，但是并没有那么做，而是用cnt[3][rt]记录每种颜色的线
条有还是没有，结果把自己卡死了。。。。
比赛后用容斥原理做了一下，好吧，还是容斥原理好做些。
建立7课线段树，拿矩形覆盖面积模版一套即可求出 R, G, B, RG, RB, GB, RGB 7种颜色
的面积值sum[]，但这个并不是最终答案，画个韦恩图就得出关系了，进而求得are[]。
注意：sum，are要 __int64
*/

int cnt[N<<2];
int len[N<<2];
int tot[10];
__int64 sum[10],are[10];

struct segment{
int lx,rx,y;
int kind;
}s[10]
;

int x[10]
;
int xcount[10];

bool cmp(segment s1,segment s2){
return s1.y<s2.y;
}

void pushup(int l,int r,int rt,int k){
if(cnt[rt])
len[rt]=x[k][r+1]-x[k][l];
else if(l==r)
len[rt]=0;
else
len[rt]=len[ls]+len[rs];
}
void updata(int ql,int qr,int l,int r,int rt,int kind,int k){
int m=(r+l)>>1;
if(l==ql && r==qr) cnt[rt]+=kind;
else if(qr<=m) updata(ql,qr,lson,kind,k);
else if(ql>m) updata(ql,qr,rson,kind,k);
else{
updata(ql,m,lson,kind,k);
updata(m+1,qr,rson,kind,k);
}
pushup(l,r,rt,k);
}

void input(int a,int x1,int x2,int y1,int y2){
x[a][tot[a]]=x1;
s[a][tot[a]].lx=x1;
s[a][tot[a]].rx=x2;
s[a][tot[a]].y=y1;
s[a][tot[a]].kind=1;
tot[a]++;

x[a][tot[a]]=x2;
s[a][tot[a]].lx=x1;
s[a][tot[a]].rx=x2;
s[a][tot[a]].y=y2;
s[a][tot[a]].kind=-1;
tot[a]++;
}

#define test(a,b) for(int ii=0;ii<(b);ii++)printf("%d%c",(a)[ii],ii==(b)-1?'\n':' ')
#define cntd(a,b) unique((a),(a)+(b))-(a)
#define lb(a,b,c) lower_bound((a),(a)+(b),(c))-(a)
#define fill(a) memset( (a),0,sizeof((a)) );

int main(){
int n,tt,col,cas=1;
scanf("%d",&tt);
while(tt--){
scanf("%d",&n);
fill(tot);
fill(sum);
fill(are);
for(int i=0;i<n;i++){
getchar();
char c;
int x1,y1,x2,y2;
scanf("%c%d%d%d%d",&c,&x1,&y1,&x2,&y2);
if(c=='R') { input(0,x1,x2,y1,y2);  input(3,x1,x2,y1,y2); input(4,x1,x2,y1,y2); input(6,x1,x2,y1,y2); }
if(c=='G') { input(1,x1,x2,y1,y2);  input(3,x1,x2,y1,y2); input(5,x1,x2,y1,y2); input(6,x1,x2,y1,y2); }
if(c=='B') { input(2,x1,x2,y1,y2);  input(4,x1,x2,y1,y2); input(5,x1,x2,y1,y2); input(6,x1,x2,y1,y2); }
}
//test(tot,7);

for(int i=0;i<7;i++){
sort(x[i],x[i]+tot[i]);
sort(s[i],s[i]+tot[i],cmp);
xcount[i]=cntd(x[i],tot[i])-1;
fill(cnt);
fill(len);
for(int j=0;j<tot[i]-1;j++){

int fx1=lb(x[i],xcount[i],s[i][j].lx);
int fx2=lb(x[i],xcount[i],s[i][j].rx)-1;
//printf("%d %d %d\n",i,fx1,fx2);
updata(fx1,fx2,0,xcount[i]-1,1,s[i][j].kind,i);
sum[i]+=(__int64)(s[i][j+1].y-s[i][j].y)*len[1];
}
}
are[0]=sum[6]-sum[5];
are[1]=sum[6]-sum[4];
are[2]=sum[6]-sum[3];
are[3]=sum[4]-sum[2]-are[0];
are[4]=sum[5]-sum[1]-are[2];
are[5]=sum[3]-sum[0]-are[1];
are[6]=sum[0]-are[0]-are[3]-are[4];

printf("Case %d:\n",cas++);
for(int i=0;i<7;i++)
printf("%I64d\n",are[i]);
}
return 0;
}