SDOI2011-染色-树链剖分

最新推荐文章于 2021-10-15 10:12:05 发布
最新推荐文章于 2021-10-15 10:12:05 发布
阅读量200 收藏
点赞数
分类专栏： 树链剖分 文章标签： 算法
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/WA_King/article/details/104118658
版权

题目

题目链接

内容

有一棵n个节点的树,每个节点有颜色,支持两种操作:

  • 把a到b这条链上的所有节点全部染成颜色c.
  • 询问a到b这条链上的颜色段数量.

分析

树上的链操作摆明了要用树链剖分.用线段树来维护信息.

线段树的每个节点需要储存,区间的颜色数量,以及区间左端点的颜色和右端点的颜色.

以下是线段树中特殊操作

如何向上更新节点

先把左右儿子的颜色段数加起来.如果左儿子的右端点和右儿子的左端点是同一种颜色,那么颜色段数-1

  void pushup(int k)
    {
        int lson=k<<1,rson=k<<1|1;
        data[k]=data[lson]+data[rson];
        cl[k]=cl[lson],cr[k]=cr[rson];
        if(cr[lson]==cl[rson]) data[k]--;
    }

如何询问区间

询问返回一个pair<int,pair<int,int>> 类型的元素, 表示区间的颜色段树,以及左右端点的颜色.

合并答案的时候同样按照向上更新节点的方式合并

  pair<int,pii> query(int l,int r,int k)
    {
        pair<int,pii> ans,now;
        ans.fi=ans.se.fi=ans.se.se=0;
        if(L[k]>=l&&R[k]<=r) 
        {
            ans.fi=data[k];
            ans.se.fi=cl[k],ans.se.se=cr[k];
            return ans;
        }
        pushdown(k);
        int mid=(L[k]+R[k])>>1;
        if(mid>=l)
        {
            ans=query(l,r,k<<1);
        }
        if(mid<r) 
        {
            now=query(l,r,k<<1|1);
            ans.fi+=now.fi;
            if(ans.se.se==now.se.fi) ans.fi--;
            if(ans.se.fi==0) ans.se.fi=now.se.fi;
            ans.se.se=now.se.se;
        }
        return ans;
    }

如何询问一条链

我们将链的端点定义问左节点和右节点.将左节点到lca定义为左链,将lca到右节点定义为左链.

分别计算左链和右链的答案,最后再合并到一起.

pair<int,pii> getsum(int l,int r)
{
    pair<int,pii> ansl,ansr,now;
    ansl.fi=ansl.se.fi=ansl.se.se=0;
    ansr.fi=ansr.se.fi=ansr.se.se=0;
    while(tp[l]!=tp[r])
    {
        if(dep[tp[l]]>dep[tp[r]])
        {
            now=tree.query(id[tp[l]],id[l],1);
            ansl.fi+=now.fi;
            if(ansl.se.fi==now.se.se) ansl.fi--;
            if(ansl.se.se==0) ansl.se.se=now.se.se;
            ansl.se.fi=now.se.fi;
            l=fa[tp[l]];
        }
        else
        {
            now=tree.query(id[tp[r]],id[r],1);
            ansr.fi+=now.fi;
            if(ansr.se.fi==now.se.se) ansr.fi--;
            if(ansr.se.se==0) ansr.se.se=now.se.se;
            ansr.se.fi=now.se.fi;
            r=fa[tp[r]];
        }
    }
    if(dep[l]>dep[r])
    {
        now=tree.query(id[r],id[l],1);
        ansl.fi+=now.fi;
        if(ansl.se.fi==now.se.se) ansl.fi--;
        if(ansl.se.se==0) ansl.se.se=now.se.se;
        ansl.se.fi=now.se.fi;
    }
    else 
    {
        now=tree.query(id[l],id[r],1);
        ansr.fi+=now.fi;
        if(ansr.se.fi==now.se.se) ansr.fi--;
        if(ansr.se.se==0) ansr.se.se=now.se.se;
        ansr.se.fi=now.se.fi;
    }
    ansl.fi+=ansr.fi;
    if(ansl.se.fi==ansr.se.fi) ansl.fi--;
    return ansl;
}

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
#define pb push_back
#define fi first
#define se second
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
const int maxn=1e5+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
int dep[maxn],son[maxn],size[maxn],tp[maxn],fa[maxn],book[maxn];
int id[maxn],rk[maxn],cnt=0;
vector<int> g[maxn];
void dfs1(int now,int f,int d)
{
    dep[now]=d,fa[now]=f,size[now]=1;
    for(auto x:g[now])
    {
        if(x==f) continue;
        dfs1(x,now,d+1);
        size[now]+=size[x];
        if(size[x]>size[son[now]]) son[now]=x;
    }
}
void dfs2(int now,int t)
{
    tp[now]=t,id[now]=++cnt,rk[cnt]=now;
    if(!son[now]) return ;
    dfs2(son[now],t);
    for(auto x: g[now])
    {
        if(x==fa[now]||x==son[now]) continue;
        dfs2(x,x);
    }
}
struct SegmentTree
{
    int data[maxn<<2],lz[maxn<<2],L[maxn<<2],R[maxn<<2],cl[maxn<<2],cr[maxn<<2];
    void pushup(int k)
    {
        int lson=k<<1,rson=k<<1|1;
        data[k]=data[lson]+data[rson];
        cl[k]=cl[lson],cr[k]=cr[rson];
        if(cr[lson]==cl[rson]) data[k]--;
    }
    void build(int l,int r,int k)
    {
        L[k]=l,R[k]=r,lz[k]=0;
        if(l==r)
        {
            data[k]=1;
            cl[k]=cr[k]=book[rk[l]];
            return ;
        }
        int mid=(l+r)>>1;
        build(l,mid,k<<1);
        build(mid+1,r,k<<1|1);
        pushup(k);
    }
    void pushdown(int k)
    {
        if(lz[k]==0) return ;
        int lson=k<<1,rson=k<<1|1;
        lz[lson]=lz[k],lz[rson]=lz[k];
        data[lson]=1,data[rson]=1;
        cl[lson]=cr[lson]=cl[rson]=cr[rson]=lz[k];
        lz[k]=0;
    }
    void change(int l,int r,int k,int x)
    {
        if(L[k]>=l&&R[k]<=r)
        {
            data[k]=1;
            cl[k]=cr[k]=x;
            lz[k]=x;
            return ;
        }
        pushdown(k);
        int mid=(L[k]+R[k])>>1;
        if(mid>=l) change(l,r,k<<1,x);
        if(mid<r) change(l,r,k<<1|1,x);
        pushup(k);
    }
    pair<int,pii> query(int l,int r,int k)
    {
        pair<int,pii> ans,now;
        ans.fi=ans.se.fi=ans.se.se=0;
        if(L[k]>=l&&R[k]<=r) 
        {
            ans.fi=data[k];
            ans.se.fi=cl[k],ans.se.se=cr[k];
            return ans;
        }
        pushdown(k);
        int mid=(L[k]+R[k])>>1;
        if(mid>=l)
        {
            ans=query(l,r,k<<1);
        }
        if(mid<r) 
        {
            now=query(l,r,k<<1|1);
            ans.fi+=now.fi;
            if(ans.se.se==now.se.fi) ans.fi--;
            if(ans.se.fi==0) ans.se.fi=now.se.fi;
            ans.se.se=now.se.se;
        }
        return ans;
    }
}tree;
void output(pair<int,pii>x)
{
    cout<<x.fi<<' '<<x.se.fi<<' '<<x.se.se<<'\n';
}
pair<int,pii> getsum(int l,int r)
{
    pair<int,pii> ansl,ansr,now;
    ansl.fi=ansl.se.fi=ansl.se.se=0;
    ansr.fi=ansr.se.fi=ansr.se.se=0;
    while(tp[l]!=tp[r])
    {
        if(dep[tp[l]]>dep[tp[r]])
        {
            now=tree.query(id[tp[l]],id[l],1);
            ansl.fi+=now.fi;
            if(ansl.se.fi==now.se.se) ansl.fi--;
            if(ansl.se.se==0) ansl.se.se=now.se.se;
            ansl.se.fi=now.se.fi;
            //cout<<l<<' '<<tp[l]<<'\n';
            //output(ansl);
            l=fa[tp[l]];
        }
        else
        {
            now=tree.query(id[tp[r]],id[r],1);
            ansr.fi+=now.fi;
            if(ansr.se.fi==now.se.se) ansr.fi--;
            if(ansr.se.se==0) ansr.se.se=now.se.se;
            ansr.se.fi=now.se.fi;
            //cout<<r<<' '<<tp[r]<<'\n';
            //output(ansr);
            r=fa[tp[r]];
        }
    }
    if(dep[l]>dep[r])
    {
        now=tree.query(id[r],id[l],1);
        ansl.fi+=now.fi;
        if(ansl.se.fi==now.se.se) ansl.fi--;
        if(ansl.se.se==0) ansl.se.se=now.se.se;
        ansl.se.fi=now.se.fi;
    }
    else 
    {
        now=tree.query(id[l],id[r],1);
        ansr.fi+=now.fi;
        //output(now);
        //output(ansr);
        if(ansr.se.fi==now.se.se) ansr.fi--;
        if(ansr.se.se==0) ansr.se.se=now.se.se;
        ansr.se.fi=now.se.fi;
        //output(ansr);
    }
    ansl.fi+=ansr.fi;
    if(ansl.se.fi==ansr.se.fi) ansl.fi--;
    return ansl;
}
void change(int l,int r,int c)
{
    while(tp[l]!=tp[r])
    {
        if(dep[tp[l]]<dep[tp[r]]) swap(l,r);
        tree.change(id[tp[l]],id[l],1,c);
        l=fa[tp[l]];
    }
    if(dep[l]>dep[r]) swap(l,r);
    tree.change(id[l],id[r],1,c);
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);
    int n,m;
    cin>>n>>m;
    rep(i,1,n) cin>>book[i];
    rep(i,1,n-1)
    {
        int u,v;
        cin>>u>>v;
        g[u].pb(v);
        g[v].pb(u);
    }
    dfs1(1,0,1);
    dfs2(1,1);
    tree.build(1,n,1);
    while(m--)
    {
        char ty;
        int a,b,c;
        cin>>ty;
        if(ty=='C')
        {
            cin>>a>>b>>c;
            change(a,b,c);
        }
        else if(ty=='Q')
        {
            cin>>a>>b;
            pair<int,pii> now;
            now=getsum(a,b);
            cout<<now.fi<<'\n';
        }
    }
    return 0;
}
WA_King
关注
专栏目录
SDOI2011 总结
qq_42835841的博客
04-22 450
文章目录2867【SDOI2011】打地鼠2868【SDOI2011】计算器2869【SDOI2011】染色2941【SDOI2011】工作安排3324【SDOI2011】黑白棋3325【SDOI2011】消防3328【SDOI2011】保密 2867【SDOI2011】打地鼠 注意到rrr和ccc无关（r⋅r\cdotr⋅c的锤子相当于用r⋅1r\cdot1r⋅1的锤子砸ccc次，ccc同理） ...
【算法竞赛学习笔记】[SDOI2011]染色（路径染色+色段查询）
SC_Linno的博客
11-20 738
title : [SDOI2011]染色（路径染色+色段查询） tags : ACM,数据结构 date : 2021-11-16 author : Linno P2486 [SDOI2011]染色 题面 给一棵树，支持两种操作： ①给定u,v,w，将u到v的路径上所有节点染成颜色w。 ②给定u,v，查询u到v路径上颜色段的数量。 树链剖分+线段树 树链剖分将无根树拍平后，用线段树去做。维护每个区间的左颜色、右颜色、色段和以及tag，如果合并的时候两端颜色一致，那么ans-1。 #include<.
SDOI2011
int128maxn的博客
10-15 164
...
SDOI2011 拦截导弹
a6t2007的博客
12-19 132
题目描述 题解： 对于第一问，我们求二维LIS即可; 对于第二问，我们可以记录向前最长长度，向前最长方案数，向后最长长度，向后最长方案数。 其实改改树状数组即可。 还有，方案数一定要开double。 代码： #include<vector> #include<cstdio> #include<cstring> #include&...
bzoj2243: [SDOI2011]染色
weixin_30794491的博客
06-22 121
2243: [SDOI2011]染色 Time Limit:20 SecMemory Limit:512 MB Submit:3271Solved:1262 [Submit][Status][Discuss] Description 给定一棵有n个节点的无根树和m个操作，操作有2类： 1、将节点a到节点b路径上所有点都染成颜色c；...
P2485 [SDOI2011]计算器
dianshu0741的博客
09-24 134
细节决定成败！！！ 这道题说上去是计算器，其实就是考你三个数论知识。 第一个操作：卡速米模板。。。 第二个操作：exgcd的运用。我并不会，这里记录一下。 对于一个\(xy \equiv z \pmod p\)，我们为了方便，换成\(ax \equiv z \pmod p\)。 根据同余的性质，得：\(ax - py = z\)，这里我们设了个\(y\)。 这种一般的二元不定方程并不知道...
【bzoj 2243】【SDOI2011】染色　题解＆代码（Ｃ＋＋）
曾经追梦的happy时光
04-14 740
bzoj SDOI 树链剖分
树链剖分模板
03-05
树链剖分模板，代码写的比较丑但是打了很多注释应该能让人看懂 HZWER流的树链剖分（倍增记录祖先） 原题是SDOI2011染色，里面有原文博客链接
树链剖分详解
wangyixin_oier的博客
08-06 255
树链剖分详解树链剖分详解树链剖分详解 转载自洛谷https://www.luogu.org/blog/communist/shu-lian-pou-fen-yang-xie 树链剖分就是将树分割成多条链，然后利用数据结构（线段树、树状数组等）来维护这些链。 别说你不知道什么是树╮(─▽─)╭（帮你百度一下） 前置知识： dfs序 LCA 线段树 先来回顾两个问题： 1，将树从 x ...
树链剖分：
_Griefs
08-24 222
这真是个神奇的东西。 一、模板例题： P3384 【模板】树链剖分： 题目描述 如题，已知一棵包含N个结点的树（连通且无环），每个节点上包含一个数值，需要支持以下操作： 操作1： 格式： 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2： 格式： 2 x y 表示求树从x到y结点最短路径上所有节点的值之和 操作3： 格式： 3 x z 表示将以x为根节点的子树内所有节点值...
[SDOI2011]消防
02-06 184
[SDOI2011]消防https://www.luogu.org/problemnew/show/P2491 题目描述 某个国家有n个城市，这n个城市中任意两个都连通且有唯一一条路径，每条连通两个城市的道路的长度为zi(zi<=1000)。 这个国家的人对火焰有超越宇宙的热情，所以这个国家最兴旺的行业是消防业。由于政府对国民的热情忍无可忍（大量的消防经费开销）可是却又无可奈何（总...
染色 SDOI2011
DREAMING的博客
07-16 293
染色 SDOI2011
【SDOI2011】消耗战
Brute♂force
04-12 311
题目 题目描述 在一场战争中，战场由 nn 个岛屿和 n-1n−1 个桥梁组成，保证每两个岛屿间有且仅有一条路径可达。现在，我军已经侦查到敌军的总部在编号为 11 的岛屿，而且他们已经没有足够多的能源维系战斗，我军胜利在望。已知在其他 kk 个岛屿上有丰富能源，为了防止敌军获取能源，我军的任务是炸毁一些桥梁，使得敌军不能到达任何能源丰富的岛屿。由于不同桥梁的材质和结构不同，所以炸毁不同的桥梁有不同...
【SDOI2011】染色
Facico的博客
07-02 1349
Description　给定一棵有n个节点的无根树和m个操作，操作有2类： 　　1、将节点a到节点b路径上所有点都染成颜色c； 　　2、询问节点a到节点b路径上的颜色段数量（连续相同颜色被认为是同一段），如“112221”由3段组成：“11”、“222”和“1”。 　　请你写一个程序依次完成这m个操作。Solution第一眼的水题这种树上的修改与询问，不是一眼的树链剖分吗！ 为了练习动态树，
[SDOI2011] 拦截导弹 （咕咕咕）
ICEEBBING的博客
05-18 202
题目 思路 有两个子任务——最多拦截的导弹个数和每个导弹被拦截的概率。 对于子任务一： 首先，我们可以定义一个dp[i]dp[i]dp[i]为拦截了第iii个导弹的最大拦截数。 易得到：dp[i]=Max(dp[j]+1) (j<i,h[j]>h[i],v[j]>v[i])dp[i]=Max(dp[j]+1)\ (j<i,h[j]>h[i],v[j]>...
[SDOI2011]染色
青烟绕指柔的博客
12-19 264
输出格式 对于每个询问操作，输出一行答案。 输入输出样例 输入 #1复制 6 5 2 2 1 2 1 1 1 2 1 3 2 4 2 5 2 6 Q 3 5 C 2 1 1 Q 3 5 C 5 1 2 Q 3 5 输出 #1复制 3 1 2 这道题真的是深有感触，当年写树剖一直过不了，也找不到bug，后来不了了之。然后现在换成LCT就好写多了，考虑的东西少了点。 我们用LCT维护链，当前节点...
2285: [Sdoi2011]保密【spfa】【最小权覆盖】
远方
03-07 761
**Description** 现在，保密成为一个很重要也很困难的问题。如果没有做好，后果是严重的。比如，有个人没有自己去修电脑，又没有拆硬盘，后来的事大家都知道了。 当然，对保密最需求的当然是军方，其次才是像那个人。为了应付现在天上飞来飞去的卫星，军事基地一般都会建造在地下。 某K国的军事基地是这样子的：地面上两排大天井共n1个作为出入口，内部是许多除可以共享出入口外互不连通的空腔
[SDOI2011]计算器
weixin_30273501的博客
08-18 225
通过我的第一篇没看题解的SD省选题祭qwq 虽然我花了2个上午才调出来 题目传送门 1.k=1时，快速幂，不会快速幂的，右转pj***。 2.k=2时，扩欧，注意要判负。 这里扯一下判负方法： 将\(a\)判负，模数为\(p\)，柿子是： \[(a \bmod p +p) \bmod p\] 3.k=3时，BSGS，我感觉我以前一直在写假的BSGSqwq 详情见代码。 #include...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值