HDU 4533 威威猫系列故事――晒被子 (线段树)

覆盖是分为两种的，从max(a1,b1)~min(a2,b2)和从min(a2,b2)~max(a2,b2)的覆盖。

所以记录两种覆盖方式对应的区间面积值。

如下图，对于第一部分的覆盖，假设有两个矩形的右上顶点分别是(a,a),(b,b)

对于第一部分的覆盖，面积为(b-x1)*(b-y1) - (a-x1)*(a-y1) = (b^2 - (x1+y1)*b) - (a^2 - (x1+y1)*a)

对于第二部分的覆盖，面积为(y2-y1)*(b-a)。

当有多个矩形时，第一种情况，二次项系数为矩形数，一次项为各矩形对应的系数相加。

第二种情况直接系数相加。

我的代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>

#define lson l , m , rt<<1
#define rson m+1,r , rt<<1|1
using namespace std;
typedef long long LL;
const LL maxn = 200010;

struct SegTree{
    LL area_1[maxn<<2],area_2[maxn<<2],val_1[maxn<<2],val_2[maxn<<2];
    LL c_1[maxn<<2],c_2[maxn<<2];

    LL func_1(LL x,LL a,LL b){
        return a*x*x - b*x;
    }
    LL func_2(LL x,LL a,LL b){
        return 1*b*x;
    }
    void push_up_1(LL rt){
        area_1[rt] = area_1[rt<<1] + area_1[rt<<1|1];
    }
    void push_up_2(LL rt){
        area_2[rt] = area_2[rt<<1] + area_2[rt<<1|1];
    }
    void push_down_1(LL l,LL r,LL rt){
        if(c_1[rt] == 0) return;
        if(l != r) c_1[rt<<1] += c_1[rt] , c_1[rt<<1|1] += c_1[rt] , val_1[rt<<1] += val_1[rt] , val_1[rt<<1|1] += val_1[rt];
        area_1[rt] += func_1(r,c_1[rt],val_1[rt]) - func_1(l-1,c_1[rt],val_1[rt]);
        c_1[rt] = val_1[rt] = 0;
    }
    void push_down_2(LL l,LL r,LL rt){
        if(c_2[rt] == 0) return;
        if(l != r) c_2[rt<<1] += c_2[rt],c_2[rt<<1|1] += c_2[rt],val_2[rt<<1] += val_2[rt],val_2[rt<<1|1] += val_2[rt];
        area_2[rt] += func_2(r,c_2[rt],val_2[rt]) - func_2(l-1,c_2[rt],val_2[rt]);
        c_2[rt] = val_2[rt] = 0;
    }
    void build(LL l,LL r,LL rt){
        area_1[rt] = area_2[rt] = val_1[rt] = val_2[rt] = c_1[rt] = c_2[rt] = 0;
        if(l == r) return;
        LL m = (l + r) >> 1;
        build(lson);build(rson);
    }
    void update_1(LL a,LL b,LL x,LL l,LL r,LL rt){
        push_down_1(l,r,rt);
        if(l > b || r < a) return;
        if(a <= l && r <= b){
            c_1[rt]++;val_1[rt] += x;
            push_down_1(l,r,rt);
            return;
        }
        LL m = (l + r) >> 1;
        update_1(a,b,x,lson);update_1(a,b,x,rson);
        push_up_1(rt);
    }
    void update_2(LL a,LL b,LL x,LL l,LL r,LL rt){
        push_down_2(l,r,rt);
        if(l > b || r < a) return;
        if(a <= l && r <= b){
            c_2[rt]++;val_2[rt] += x;
            push_down_2(l,r,rt);
            return;
        }
        LL m = (l + r) >> 1;
        update_2(a,b,x,lson);update_2(a,b,x,rson);
        push_up_2(rt);
    }
    LL query(LL a,LL b,LL l,LL r,LL rt){
        push_down_1(l,r,rt),push_down_2(l,r,rt);
        if(l > b || r < a) return 0;
        if(a <= l && r <= b) return area_1[rt] + area_2[rt];
        LL m = (l + r) >> 1;
        return query(a,b,lson) + query(a,b,rson);
    }
}sol;

LL n,q,ax[maxn],ay[maxn],bx[maxn],by[maxn],t[maxn],o;

int main(){
    LL a,b,c,cas;
    scanf("%I64d",&cas);
    while(cas--){
        scanf("%I64d",&n);o = 0;
        for(int i=0;i<n;i++){
            scanf("%I64d%I64d%I64d%I64d",&ax[i],&ay[i],&bx[i],&by[i]);
            o = max(o,ax[i]);o = max(o,ay[i]);
            o = max(o,bx[i]);o = max(o,by[i]);
        }
        scanf("%I64d",&q);
        for(int i=0;i<q;i++){
            scanf("%I64d",&t[i]);o = max(o,t[i]);
        }
        sol.build(1,o,1);
        for(int i=0;i<n;i++){
            a = max(ax[i],ay[i]);b = min(bx[i],by[i]);c = max(bx[i],by[i]);
            if(a<b) sol.update_1(a+1,b,ax[i]+ay[i],1,o,1);
            if(bx[i]<by[i]) sol.update_2(max(bx[i],ay[i])+1,by[i],bx[i]-ax[i],1,o,1);
            if(by[i]<bx[i]) sol.update_2(max(by[i],ax[i])+1,bx[i],by[i]-ay[i],1,o,1);
        }
        for(int i=0;i<q;i++){
            printf("%I64d\n",sol.query(1,t[i],1,o,1));
        }
    }

    return 0;
}