POJ 2777 Count Color

最新推荐文章于 2021-08-12 22:39:07 发布

嘿嘿不错

最新推荐文章于 2021-08-12 22:39:07 发布

阅读量276

点赞数

分类专栏： acm

本文链接：https://blog.csdn.net/ACMer_AK/article/details/51971310

版权

acm 专栏收录该内容

62 篇文章 0 订阅

订阅专栏

题意：L的长度，从1~L，在区间上添加颜色（1~T），颜色初始全为1。输入P A B ：查询A~B区间不同颜色的数目；输入C A B C：更新A~B区间的颜色为C；

运用线段树的延时标记。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#define MAXN 100010
using namespace std;

struct node{
    int l,r;
    int color;
}a[MAXN<<2];

int vis[40];
int ans = 0;

void build(int l,int r,int index){
    a[index].l = l;
    a[index].r = r;
    a[index].color = 1;//一定要初始为1；
    if(l == r){
        return ;
    }
    int mid = (l+r)>>1;
    build(l,mid,index<<1);
    build(mid+1,r,index<<1|1);
}

void updata(int l,int r,int index,int color){

    if(l <= a[index].l && r >= a[index].r){
        a[index].color = color;
        return ;
    }
    if(a[index].color){
        a[index<<1].color = a[index].color;
        a[index<<1|1].color = a[index].color;
        a[index].color = 0;
    }
    int mid = (a[index].l+a[index].r)>>1;
    if(l <= mid){
        updata(l,r,index<<1,color);
    }
    if(r > mid){
        updata(l,r,index<<1|1,color);
    }
}

void query(int l,int r,int index){
    if(r < a[index].l || l > a[index].r){
        return ;
    }
    //L 为a[index].l
    //R 为a[index].r
    //1: l L r R
    //2: L l R r
    //3: L l r R
    //4: l L R r
    if((l >= a[index].l || r <= a[index].r || (l <= a[index].l && r >= a[index].r))&& a[index].color){
        if(!vis[a[index].color]){
            ans++;
            vis[a[index].color] = 1;
        }
        return ;
    }
    int mid = (a[index].l+a[index].r)>>1;
    if(l <= mid){
        query(l,r,index<<1);
    }
    if(r > mid){
        query(l,r,index<<1|1);
    }
    return ;
}

int main(){
    int l,t,o;
    int a,b,c;
    char s[2];
    scanf("%d %d %d",&l,&t,&o);
    build(1,l,1);
    for(int i = 0; i < o; i++){
        scanf("%s",s);
        if(!strcmp(s,"C")){
            scanf("%d %d %d",&a,&b,&c);
            if(a > b){
                swap(a,b);
            }
            updata(a,b,1,c);
        }
        else{
            scanf("%d %d",&a,&b);
            if(a > b){
                swap(a,b);
            }
            memset(vis,0,sizeof(vis));
            ans = 0;
            query(a,b,1);
            printf("%d\n",ans);
        }
    }
    return 0;
}