HDU 4419 Colourful Rectangle --离散化+线段树扫描线

题意： 有三种颜色的矩形n个，不同颜色的矩形重叠会生成不同的颜色，总共有R,G,B,RG,RB,GB,RGB 7种颜色，问7种颜色每种颜色的面积。

解法： 很容易想到线段树扫描线求矩形面积并，但是如何维护每种颜色的长度着实让我伤透了脑筋。后来看了一位朋友的题解，才幡然醒悟。

开始想到了用二进制表示颜色，R用001表示，G用010表示，B用100表示。那么就可以用十进制1~7表示7种不同颜色了。

维护 cov[rt][1~3] 表示此区间内3种原色各有多少个， Len[rt][i]表示每种颜色的长度。

重点是pushup的写法，详情见代码， 离散化什么还比较简单。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#define lll __int64
using namespace std;
#define N 10107

lll Len[*N][];
int cov[*N][];

struct Line
{
    int y1,y2,x;
    int col;
}line[*N];
int yy[*N];

int cmp(Line ka,Line kb) { return ka.x < kb.x; }

void pushup(int l,int r,int rt)
{
    int state = ;
    for(int i=;i<=;i++) if(cov[rt][i]) state |= (<<(i-));  //表示此区间内有的颜色
    for(int i=;i<;i++)  Len[rt][i] = ;                      //加了一层，要重新算各个颜色的长度
    if(l+ == r)                                               //叶子节点
    {
        Len[rt][state] = yy[r]-yy[l];                          //只有此颜色state，长度为yy[r]-yy[l]
        return;
    }
    for(int i=;i<;i++)                                       //否则使用下面的更新
        Len[rt][i|state] += Len[*rt][i] + Len[*rt+][i];
}

void build(int l,int r,int rt)
{
    for(int i=;i<=;i++) cov[rt][i] = ;
    for(int i=;i<;i++)  Len[rt][i] = ;
    Len[rt][] = yy[r]-yy[l];
    if(l == r-) return;
    int mid = (l+r)/;
    build(l,mid,*rt);
    build(mid,r,*rt+);
}

void update(int l,int r,int aa,int bb,int cover,int rt)
{
    if(aa <= l && bb >= r)
    {
        if(cover > ) cov[rt][cover]++;
        else          cov[rt][-cover]--;
        pushup(l,r,rt);
        return;
    }
    if(l+ == r) return;
    int mid = (l+r)/;
    if(aa <= mid)
        update(l,mid,aa,bb,cover,*rt);
    if(bb > mid)
        update(mid,r,aa,bb,cover,*rt+);
    pushup(l,r,rt);
}

int main()
{
    int n,m,t,cs = ,i,j;
    int x1,y1,x2,y2,c;
    char ss[];
    scanf("%d",&t);
    while(t--)
    {
        m = ;
        scanf("%d",&n);
        for(i=;i<n;i++)
        {
            scanf("%s%d%d%d%d",ss,&x1,&y1,&x2,&y2);
            if(ss[] == 'R')      c = ;
            else if(ss[] == 'G') c = ;
            else                  c = ;
            line[m].x = x1,line[m].y1 = y1,line[m].y2 = y2,line[m].col = c,yy[m++] = y1;
            line[m].x = x2,line[m].y1 = y1,line[m].y2 = y2,line[m].col = -c,yy[m++] = y2;
        }
        m--;
        sort(yy+,yy+m+);
        int cnt = ;
        for(i=;i<=m;i++)
        {
            if(yy[i] != yy[i-])
                yy[cnt++] = yy[i];
        }
        cnt--;
        build(,cnt,);
        sort(line+,line+m+,cmp);
        lll ans[];
        memset(ans,,sizeof(ans));
        printf("Case %d:\n",cs++);
        for(i=;i<m;i++)
        {
            int L = lower_bound(yy+,yy+cnt+,line[i].y1)-yy;
            int R = upper_bound(yy+,yy+cnt+,line[i].y2)-yy-;
            update(,cnt,L,R,line[i].col,);
            for(j=;j<;j++)
                ans[j] += Len[][j]*(line[i+].x-line[i].x);
        }
        printf("%I64d\n%I64d\n%I64d\n%I64d\n%I64d\n%I64d\n%I64d\n",ans[],ans[],ans[],ans[],ans[],ans[],ans[]);
    }
    return ;
}