ZOJ 1610 Count the Colors 线段树区间染色

题目的意思就是给你在区间内染色，后染色的可以覆盖前染色的，自己写的时候写炸了，没想通是区间染色，也就是在[L,R]之间染色，不是点染色，所以为了区分点和区间，我们把[L,R]的区间染色变成[L+1,R]的点染色，这个操作自己刚开始想到了，就是在计数的时候自己出现了问题，计数时怎么判断前面那个数和上次的数是不是一个数就可以了，不用判断每种颜色上一次染色时候的R+1和这一次染色的L是不是相等。第一次写的是将线段树更新到叶子节点再计数，然后自己改进了一下把当前节点当做lazy标记，然后从顶到底的查询，最重要的是当L==R时要退出，否则会死循环，且当区间之间断开时，要标注将前一个标记的数变成-1

更新到叶子结点：

#include<bits/stdc++.h>
using namespace std;
using LL = int64_t;
const int maxn=8005;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const LL LLINF=1e14;
const double pi=acos(-1.0);
int sum[maxn],tree[maxn*4],pre=-1;

void push_down(int num){
    tree[num<<1]=tree[num<<1|1]=tree[num];
    tree[num]=-1;
}

void update(int num,int l,int r,int _l,int _r,int d) {
    if(_l<=l&&_r>=r) {
        tree[num]=d;
        return;
    }
    if(tree[num]!=-1)push_down(num);
    int mid=(l+r)>>1;
    if(_l<=mid) update(num<<1,l,mid,_l,_r,d);
    if(_r>mid) update(num<<1|1,mid+1,r,_l,_r,d);
}

void dfs(int num,int l,int r) {
    if(l==r) {
        if(tree[num]!=-1&&pre!=tree[num]) sum[tree[num]]++;
        pre=tree[num];
        return ;
    }
    if(tree[num]!=-1)push_down(num);
    int mid=(l+r)>>1;
    dfs(num<<1,l,mid);
    dfs(num<<1|1,mid+1,r);
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n;
    while(cin>>n) {
        memset(sum,0,sizeof(sum));
        memset(tree,-1,sizeof(tree));
        int l,r,d;
        for(int i=1;i<=n;i++) {
            cin>>l>>r>>d;
            update(1,1,8000,l+1,r,d);
        }
        pre=-1;
        dfs(1,1,8000);
        for(int i=0;i<=8000;i++) {
            if(sum[i]) cout<<i<<" "<<sum[i]<<endl;
        }
        cout<<endl;
    }
    return 0;
}

改进后：

#include<bits/stdc++.h>
using namespace std;
using LL = int64_t;
const int maxn=8005;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const LL LLINF=1e14;
const double pi=acos(-1.0);
int sum[maxn],tree[maxn*4],pre=-1;

void push_down(int num){
    tree[num<<1]=tree[num<<1|1]=tree[num];
    tree[num]=-1;
}

void update(int num,int l,int r,int _l,int _r,int d) {
    if(_l<=l&&_r>=r) {
        tree[num]=d;
        return;
    }
    if(tree[num]!=-1)push_down(num);
    int mid=(l+r)>>1;
    if(_l<=mid) update(num<<1,l,mid,_l,_r,d);
    if(_r>mid) update(num<<1|1,mid+1,r,_l,_r,d);
}

void dfs(int num,int l,int r) {
    if(tree[num]!=-1) {
        if(pre!=tree[num]) sum[tree[num]]++;
        pre=tree[num];
        return ;
    }
    if(l==r) {pre=-1;return ;}
    int mid=(l+r)>>1;
    dfs(num<<1,l,mid);
    dfs(num<<1|1,mid+1,r);
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n;
    while(cin>>n) {
        memset(sum,0,sizeof(sum));
        memset(tree,-1,sizeof(tree));
        int l,r,d;
        for(int i=1;i<=n;i++) {
            cin>>l>>r>>d;
            update(1,1,8000,l+1,r,d);
        }
        pre=-1;
        dfs(1,1,8000);
        for(int i=0;i<=8000;i++) {
            if(sum[i]) cout<<i<<" "<<sum[i]<<endl;
        }
        cout<<endl;
    }
    return 0;
}