Luogu P1558 色板游戏

Posted 2022-09-03 theshadow

闲扯

一道线段树好题裸题

顺便吐槽一下出题人的数据。。。

最开始没看到 \(A\) 可能大于 \(B\) ，结果 \(WA\) 了。。

题面

题面

Solution

看范围， \(T\leq30\) ，发现可以用二进制表示是否包含每一种颜料，接下来的事就好办了。

对于操作 \(1\) ，是一个区间覆盖，直接将 \(s\) 变成 \(1<<(C-1)\) ，表示这个区间现在只含有 \(C\) 这种颜料，懒标记换一下。

下放也是一样的操作。

向上统计的时候直接用左区间的 \(s\) 或上右区间的 \(s\) 即 \(T[cur].s=T[lc].s\mid T[rc].s\) 。

每一次查询时找到表示这段区间的 \(s\) ，然后直接查看有包含几个 \(1\) 即可。

Code

#include<bits/stdc++.h>
#define del(a,i) memset(a,i,sizeof(a))
#define ll long long
#define inl inline
#define il inl void
#define it inl int
#define ill inl ll
#define re register
#define ri re int
#define rl re ll
#define mid ((l+r)>>1)
#define lowbit(x) (x&(-x))
#define INF 0x3f3f3f3f
using namespace std;
template<class T>il read(T &x)
    int f=1;char k=getchar();x=0;
    for(;k>'9'||k<'0';k=getchar()) if(k=='-') f=-1;
    for(;k>='0'&&k<='9';k=getchar()) x=(x<<3)+(x<<1)+k-'0';
    x*=f;

template<class T>il print(T x)
    if(x/10) print(x/10);
    putchar(x%10+'0');

ll mul(ll a,ll b,ll mod)long double c=1.;return (a*b-(ll)(c*a*b/mod)*mod)%mod;
it qpow(int x,int m,int mod)
    int res=1,bas=x%mod;
    while(m)
        if(m&1) res=(res*bas)%mod;
        bas=(bas*bas)%mod,m>>=1;
    
    return res%mod;

const int MAXN = 1e5+5;
int n,m,k,x,y,z;
char opt;
#define lc (cur<<1)
#define rc (cur<<1|1)
struct Seg_Tree
    int tag,s;
T[MAXN<<2];
il pushup(int cur)T[cur].s=T[lc].s|T[rc].s;
il pushdown(int cur,int l,int r)
    if(!T[cur].tag) return ;
    T[lc].tag=T[rc].tag=T[cur].tag;
    T[lc].s=T[rc].s=1<<(T[cur].tag-1);
    T[cur].tag=0;

il build_tree(int cur,int l,int r)
    T[cur].s=1;
    if(l==r) return ;
    build_tree(lc,l,mid),build_tree(rc,mid+1,r);

it query(int cur,int l,int r,int L,int R)
    if(l>=L&&r<=R) return T[cur].s;
    pushdown(cur,l,r);ri res=0;
    if(mid>=L) res|=query(lc,l,mid,L,R);
    if(R>mid) res|=query(rc,mid+1,r,L,R);
    return res;

il updata(int cur,int l,int r,int L,int R,int k)
    if(l>=L&&r<=R) T[cur].s=1<<(k-1),T[cur].tag=k;
    else
        pushdown(cur,l,r);
        if(mid>=L) updata(lc,l,mid,L,R,k);
        if(R>mid) updata(rc,mid+1,r,L,R,k);
        pushup(cur);
    

it count(int x)
    ri res=0;
    while(x)
        if(x&1) res++;
        x>>=1;
    
    return res;

int main()

//  freopen("testdata.in","r",stdin);
//  freopen("1.out","w",stdout);
    read(n),read(k),read(m);
    build_tree(1,1,n);
    for(ri i=1;i<=m;++i)
        cin>>opt;read(x),read(y);
        if(x>y) swap(x,y);
        if(opt=='C') read(z),updata(1,1,n,x,y,z);
        else
            ri tmp=query(1,1,n,x,y);
            print(count(tmp)),puts("");
        
    
    return 0;

总结

蒟蒻颓了一晚上还是做点题吧。。

马上 \(NOIP\) 了虽然它已经死了，感觉好慌呀 \(qwq\) 。

这样的模板题要追求速度，同时还要保证正确率，不然考场上一慌就 \(GG\) 。