P1558 色板游戏

题目链接

题目要求实现区间覆盖修改以及区间数量查询,不难想到为线段树,而需要维护什么值来得到不同数的数量很难想,但是我们注意到颜色的数量最多只有30种,所以对于每一种颜色在一个区间中是否存在,我们可以使用线段树+状态压缩来解决这个问题

首先考虑pushup,这点很简单,只要将两个儿子节点的颜色状态或一下就可以

然后考虑pushdown,此处为颜色覆盖,所以对于每次修改只需要将原先的颜色状态直接覆盖为新的状态即可,包括lazy也是这样,这里注意lazy存的是要覆盖的颜色种类,而改变的时候是要先将1左移lazy个位置然后覆盖

ac代码:

cpp 复制代码
#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define INF 0x3f3f3f3f
#define pb push_back
#define int long long
#define Mirai ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;
typedef pair<int,int> pii;
const int N=1e5+10;
struct Tree
{
    int l,r;
    int sum,lazy;
}tr[N<<2];
int n,m,q;
string op;
int l,r,d;
int lowbit(int x){return x&-x;}
void change(int u,int lazy)
{
    tr[u].sum=1<<lazy;
    tr[u].lazy=lazy;
}
void pushup(int u)
{
    tr[u].sum=tr[u<<1].sum|tr[u<<1|1].sum;
}
void pushdown(int u)
{
    if(tr[u].lazy)
    {
        change(u<<1,tr[u].lazy);
        change(u<<1|1,tr[u].lazy);
        tr[u].lazy=0;
    }
}
void build(int u,int l,int r)
{
    if(l==r)tr[u]={l,r,1<<1,0};
    else 
    {
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
        pushup(u);
    }
}
void modify(int u,int l,int r,int d)
{
    if(tr[u].l>=l&&tr[u].r<=r)
    {
        change(u,d);
        return ;
    }
    pushdown(u);
    int mid=tr[u].l+tr[u].r>>1;
    if(l<=mid)modify(u<<1,l,r,d);
    if(r>mid)modify(u<<1|1,l,r,d);
    pushup(u);
}
int query(int u,int l,int r)
{
    if(tr[u].l>=l&&tr[u].r<=r)return tr[u].sum;
    pushdown(u);
    int mid=tr[u].l+tr[u].r>>1;
    int res=0;
    if(l<=mid)res|=query(u<<1,l,r);
    if(r>mid)res|=query(u<<1|1,l,r);
    return res;
}
void solve()
{
    cin>>n>>m>>q;
    build(1,1,n);
    while(q--)
    {
        cin>>op>>l>>r;
        if(l>r)swap(l,r);
        if(op=="C")
        {
            cin>>d;
            modify(1,l,r,d);
        }
        else 
        {
            int ans=query(1,l,r);
            int cnt=0;
            while(ans)
            {
                cnt++;
                ans-=lowbit(ans);
            }
            cout<<cnt<<endl;
        }
    }
}
signed main()
{
    Mirai;
    int T=1;
    //cin>>T;
    while(T--)
    {
        solve();
    }
}
相关推荐
BingLin-Liu3 分钟前
图论之cruskal算法(克鲁斯卡尔)
数据结构·算法·图论
进取星辰17 分钟前
PyTorch 深度学习实战(15):Twin Delayed DDPG (TD3) 算法
pytorch·深度学习·算法
strive-debug22 分钟前
C语言之 条件编译和预处理指令
算法
Sacuki1 小时前
BP神经网络公式推导与代码实现
算法
山登绝顶我为峰 3(^v^)31 小时前
VSCode + CMake
ide·vscode·算法·计算机·编辑器
勇敢滴勇2 小时前
【C++】二叉搜索树(二叉查找树、二叉排序树)详解
开发语言·c++·算法·霍夫曼树
f狐0狸x2 小时前
【蓝桥杯每日一题】3.16
c++·算法·蓝桥杯
周Echo周2 小时前
8、STL中的map和pair使用方法
开发语言·数据结构·c++·考研·算法·leetcode·pat考试
IT猿手2 小时前
基于牛优化( OX Optimizer,OX)算法的多个无人机协同路径规划(可以自定义无人机数量及起始点),MATLAB代码
人工智能·算法·机器学习·matlab·无人机
武乐乐~3 小时前
欢乐力扣:反转链表
算法·leetcode·链表