【JZOJ4603】颜料大乱斗

来源：互联网发布：手机弹电子琴软件编辑：程序博客网时间：2024/04/25 22:39

Description

这里写图片描述

Solution

辣鸡题目！！！

(这里就不骂辣鸡出题人了)

这题查询和修改区间(l,r)竟然l可以大于r！！

而且白色是颜料1！！！！

注意上面那些细节，开30颗线段树就可以AC了。

Code

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#define fo(i,j,k) for(int i=j;i<=k;i++)#define fd(i,j,k) for(int i=j;i>=k;i--)#define N 100001using namespace std;bool tr[N*4][31];int g[N*4];int a[N];bool bz[31];int n,c,m;void put(int v,int l,int r){    if(!g[v]) return;    fo(i,1,c) tr[v*2][i]=tr[v*2+1][i]=0;    tr[v*2][g[v]]=1;    tr[v*2+1][g[v]]=1;    g[v*2]=g[v*2+1]=g[v];    g[v]=0;}void build(int v,int l,int r){    if(l==r)    {        tr[v][1]=1;        g[v]=0;        return;    }    int mid=(l+r)/2;    build(v*2,l,mid);    build(v*2+1,mid+1,r);    tr[v][1]=tr[v*2][1]|tr[v*2+1][1];    g[v]=0;}void change(int v,int l,int r,int x,int y,int t){    if(l==x && r==y)    {        g[v]=t;        fo(i,1,c) tr[v][i]=0;        tr[v][t]=1;        return;    }    put(v,l,r);    int mid=(l+r)/2;    if(y<=mid) change(v*2,l,mid,x,y,t);    else if(x>mid) change(v*2+1,mid+1,r,x,y,t);    else    {        change(v*2,l,mid,x,mid,t);        change(v*2+1,mid+1,r,mid+1,y,t);    }    fo(i,1,c)    tr[v][i]=tr[v*2][i]|tr[v*2+1][i];}int ans=0;void find(int v,int l,int r,int x,int y){    if(l==x && r==y)    {        fo(i,1,c)        if(tr[v][i] && !bz[i]) bz[i]=true,ans++;        return;    }    put(v,l,r);    int mid=(l+r)/2;    if(y<=mid) find(v*2,l,mid,x,y);    else if(x>mid) find(v*2+1,mid+1,r,x,y);    else    {        find(v*2,l,mid,x,mid);        find(v*2+1,mid+1,r,mid+1,y);    }}int main(){    cin>>n>>c>>m;    build(1,1,n);    while(m--)    {        char ch[3];        int l,r;        scanf("%s %d %d",ch,&l,&r);        if(l>r) swap(l,r);        if(ch[0]=='C')        {            int t;            scanf("%d",&t);            change(1,1,n,l,r,t);        }        else        {            memset(bz,0,sizeof(bz));            ans=0;            find(1,1,n,l,r);            if(!ans) ans++;            printf("%d\n",ans);        }    }}

0 0