【JZOJ 4603】颜料大乱斗

Description

画师傅又要开始画画了，这次他花了三天三夜将一堵墙涂成了白色。
但是画师傅有个顽劣的弟子小花，小花讨厌画师傅对Ta始乱终弃，所以趁 画师傅不在用不同的颜料将墙涂来涂去。
然而画师傅为了保护他的大作，设置了一个监控机制A。A每隔一段时间会 查珣某一段墙上的颜料的种类数，并将其记录下来。
现在画师傅回来了，看到五颜六色色彩斑斓灯火阑珊金碧辉煌的【哗】墙， 一口气没换上来，卒。
作为画师傅的好友杀小姐的你，为了追查画师傅的死因，于是调用了墙的监 控机制A的记录。于是，你看到了怎样的记录呢？

Solution

很显然，开30棵线段树即可，
但是：
有可能l>r啊！！！
白色是1啊！！！
复杂度：$O(n*30\log(n))$；
要优化常数。

Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#define fo(i,a,b) for(int i=a;i<=b;i++) 
using namespace std;
const int N=100500;
int read(int &n)
{
    char ch=' ';int q=0,w=1;
    for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
    if(ch=='-')w=-1,ch=getchar();
    for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;n=q*w;return n;
}
int n,m,ans;
int b[N*3][32],la[N*3];
int s[32];
void doit(int l,int r,int e)
{
    if(!la[e])return;
    fo(i,0,m)b[e][i]=0;
    b[e][la[e]]=r-l+1;
    if(l!=r)la[e*2]=la[e],la[e*2+1]=la[e];
    la[e]=0;
}
void build(int l,int r,int e)
{
    if(l==r)
    {
        b[e][1]=1;
        return;
    }
    int t=(l+r)/2;
    build(l,t,e*2);
    build(t+1,r,e*2+1);
    b[e][1]=r-l+1;
}
void change(int l,int r,int e,int l1,int r1,int l2)
{
    if(l==l1&&r==r1)
    {
        la[e]=l2;
        doit(l,r,e);
        return;
    }
    int t=(l+r)/2;
    doit(l,t,e*2),doit(t+1,r,e*2+1);
    if(r1<=t)change(l,t,e*2,l1,r1,l2);
        else if(t<l1)change(t+1,r,e*2+1,l1,r1,l2);
            else change(l,t,e*2,l1,t,l2),change(t+1,r,e*2+1,t+1,r1,l2);
    fo(i,0,m)b[e][i]=b[e*2][i]+b[e*2+1][i];
}
void find(int l,int r,int e,int l1,int r1)
{
    doit(l,r,e);
    if(l==l1&&r==r1)
    {
        fo(i,0,m)s[i]+=b[e][i];
        return;
    }
    int t=(l+r)/2;
    if(r1<=t)find(l,t,e*2,l1,r1);
        else if(t<l1)find(t+1,r,e*2+1,l1,r1);
            else find(l,t,e*2,l1,t),find(t+1,r,e*2+1,t+1,r1);
}
int main()
{
    int q,l,r,m_;
    read(n),read(m),read(m_);
    build(1,n,1);
    while(m_--)
    {
        char ch=' ';
        while(ch!='P'&&ch!='C')ch=getchar();
        read(l),read(r);
        if(r<l)swap(l,r);
        if(ch=='C')
        {
            read(q);
            change(1,n,1,l,r,q);
        }
        else 
        {
            ans=q=0;
            fo(i,0,m)s[i]=0;
            find(1,n,1,l,r);
            fo(i,1,m)if(s[i])ans++,q+=s[i];
            printf("%d\n",ans+(q<r-l+1));
        }
    }
    return 0;
}