【jzoj4603】【颜料大乱斗】【线段树】

题目大意

给出一个长度为n的区间，有一些操作，对一个区间染上某种颜色，询问一个区间有多少种颜色。

解题思路

题目所说，颜色最多有30种，注意初始颜色是1，显然可以开30棵线段树，处理每种颜色。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=100000,maxc=30;
int n,m,c,ans,pon=maxc+1,tag[maxn*4+10],flag[30+10];
bool size[maxc+1][maxn*4+10];
LL read(){
    LL val=0;char ch=getchar();
    for(;(ch<'0')||(ch>'9');ch=getchar());
    for(;(ch>='0')&&(ch<='9');val=val*10+ch-'0',ch=getchar());
    return val;
}
char readch(){
    char ch=getchar();
    for(;(ch!='C')&&(ch!='P');ch=getchar());
    return ch;
}
void pushtag(int now){
    int ls=now<<1,rs=(now<<1)+1;
    tag[ls]=tag[rs]=tag[now];
    fo(i,1,30)
        size[i][ls]=size[i][rs]=(i==tag[now]);
    tag[now]=0;
}
void query(int now,int l,int r,int l1,int r1){
    int m=(l+r)>>1,ls=now<<1,rs=(now<<1)+1;
    if(tag[now]&&(l!=r))pushtag(now);
    if((l==l1)&&(r==r1)){
        fo(i,1,30)
            if((!flag[i])&&size[i][now]){
                ans++;
                flag[i]=1;
            }
    }
    else if(r1<=m)query(ls,l,m,l1,r1);
    else if(m<l1)query(rs,m+1,r,l1,r1);
    else{
        query(ls,l,m,l1,m);
        query(rs,m+1,r,m+1,r1);
    }
}
void change(int now,int l,int r,int l1,int r1,int c){
    int m=(l+r)>>1,ls=now<<1,rs=(now<<1)+1;
    if(tag[now]&&(l!=r))pushtag(now);
    if((l==l1)&&(r==r1)){
        tag[now]=c;
        fo(i,1,30)
            size[i][now]=(c==i);
        return;
    }
    else if(r1<=m)change(ls,l,m,l1,r1,c);
    else if(m<l1)change(rs,m+1,r,l1,r1,c);
    else{
        change(ls,l,m,l1,m,c);
        change(rs,m+1,r,m+1,r1,c);
    }
    fo(i,1,30)size[i][now]=size[i][ls]||size[i][rs];
}
int main(){
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    //scanf("%d%d%d\n",&n,&c,&m);
    n=read();c=read();m=read();
    change(1,1,n,1,n,1);
    fo(i,1,m){
        char ch=readch();//scanf("%c",&ch);
        if(ch=='P'){
            int x,y,xx,yy;xx=read();yy=read();//scanf("%d%d\n",&x,&y);
            ans=0;memset(flag,0,sizeof(flag));x=min(xx,yy);y=max(xx,yy);
            query(1,1,n,x,y);
            printf("%d\n",ans);
        }else{
            int x,y,xx,yy,t;xx=read();yy=read();t=read();//scanf("%d%d%d\n",&x,&y,&t);
            x=min(xx,yy);y=max(xx,yy);
            change(1,1,n,x,y,t);
        }
    }
    return 0;
}