ZOJ 1610 Count the Colors 【线段树】

<题目链接>

题目大意：

在[0,8000]这个区间内，不断进行一些操作，将其中的一些区间染成特定颜色，如果区间重复的话，后面染的色块会覆盖前面染的色块，问最终[0,8000]这个区间内每种颜色的色块数量是多少。

解题分析：

首先要注意，这是对区间进行更新，。所以update的时候是对输入区间[a,b]的左闭右开或者是左开右闭进行处理。然后本题思路非常清晰，就是按照输入顺序更新线段树对应区间，然后对[0,8000]这个区间从左向右进行色块的统计。PS:虽然最后还是要将所有lazy下放到所有的叶子节点，进行色块数量的暴力统计，但是pushdown函数确实能够减少 update 的一些复杂度。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define Lson rt<<1,l,mid
#define Rson rt<<1|1,mid+1,r 
const int M =8e3;
int tr[M<<2],lazy[M<<2],last;
int col[M<<1];
void Pushdown(int rt){
    if(lazy[rt]!=-1){
        int tmp=lazy[rt];
        lazy[rt<<1]=lazy[rt<<1|1]=tmp;
        tr[rt<<1]=tr[rt<<1|1]=tmp;
        lazy[rt]=-1;
    }
}
void update(int rt,int l,int r,int L,int R,int c){  //区间染色
    if(L<=l&&r<=R){
        lazy[rt]=c;
        tr[rt]=c;
        return;
    }
    Pushdown(rt);
    int mid=(l+r)>>1;
    if(L<=mid)update(Lson,L,R,c);
    if(R>mid)update(Rson,L,R,c);
}
void query(int rt,int l,int r){  //运用递归（类似于dfs），达到从左到右逐步查询的效果，当遇到染色的点时，判断其是否与前一个点相同，如果不同，就说明该颜色的区块
    if(l==r){
        if(tr[rt]!=last&&tr[rt]!=-1)col[tr[rt]]++;
        last=tr[rt];
        return;
    }
    Pushdown(rt);
    int mid=(l+r)>>1;
    query(Lson);
    query(Rson);
}
int main(){
    int n;
    while(scanf("%d",&n)!=EOF){
        memset(tr,-1,sizeof(tr));
        memset(lazy,-1,sizeof(lazy));
        memset(col,0,sizeof(col));
        while(n--){
            int x,y,c;
            scanf("%d%d%d",&x,&y,&c);
            update(1,1,M,x+1,y,c);  //由于本题是区间更新，所以不能直接更新[a,b]这个闭区间的所有点,而是对a~b的左开右闭或者左闭右开区间进行更新,类似于图的将边权转化为点权
        }
        last=-1;
        query(1,1,M);
        for(int i=0;i<=M;i++){
            if(col[i])printf("%d %d\n",i,col[i]);
        }
        printf("\n");
    }
    return 0;
}

 

 

2018-09-22