[SDOI2011]染色(luogu)

Description

给定一棵有n个节点的无根树和m个操作,操作有2类:
1、将节点a到节点b路径上所有点都染成颜色c;
2、询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段),
如“112221”由3段组成:“11”、“222”和“1”。
请你写一个程序依次完成这m个操作。
input
第一行包含2个整数n和m,分别表示节点数和操作数;
第二行包含n个正整数表示n个节点的初始颜色
下面 行每行包含两个整数x和y,表示x和y之间有一条无向边。
下面 行每行描述一个操作:
“C a b c”表示这是一个染色操作,把节点a到节点b路径上所有点(包括a和b)都染成颜色c;
“Q a b”表示这是一个询问操作,询问节点a到节点b(包括a和b)路径上的颜色段数量。
ouput
对于每个询问操作,输出一行答案。
Solution
树链剖分处理色块(注意相邻两段的首尾颜色相同的情况)
主要是细节,详见代码
Code

#include <cstdio>
#include <cstdlib>
#include <vector>
using namespace std;
const int N=1e5+;
struct node
{
int l,r,lc,rc,cl,cr,sum;
bool flag;
}f[N*];
int fa[N],deep[N],son[N],d[N],dfn[N],si[N],top[N],rt,
tot,cnt,n,m,u,v,c,re[N],L,R,a,b;
char s[];
vector <int> link[N];
void dfs1(int u,int f)
{
fa[u]=f;
deep[u]=deep[f]+;
si[u]=;
int size=link[u].size();
for(int i=;i<size;i++)
{
int v=link[u][i];
if(v!=f)
{
dfs1(v,u);
si[u]+=si[v];
if(son[u]== || si[son[u]]<si[v]) son[u]=v;
}
}
}
void dfs2(int u,int f)
{
dfn[u]=++cnt,re[cnt]=u;
if(son[u]!=) top[son[u]]=top[u],dfs2(son[u],u);
int size=link[u].size();
for(int i=;i<size;i++)
{
int v=link[u][i];
if(v!=f && v!=son[u])
top[v]=v,dfs2(v,u);
}
}
void push_up(int g)
{
int lc=f[g].lc,rc=f[g].rc;
f[g].cl=f[lc].cl,f[g].cr=f[rc].cr;
f[g].sum=f[lc].sum+f[rc].sum;
if(f[lc].cr==f[rc].cl) f[g].sum--;
}
void push_down(int g)
{
if(f[g].flag)
{
int lc=f[g].lc,rc=f[g].rc;
f[lc].flag=f[rc].flag=true;
f[lc].cl=f[lc].cr=f[rc].cl=f[rc].cr=f[g].cl;
f[lc].sum=f[rc].sum=;
f[g].flag=false;
}
}
void build(int &g,int l,int r)
{
g=++tot;
f[g].l=l,f[g].r=r;
if(l==r)
{
f[g].flag=true,f[g].sum=;
f[g].cl=f[g].cr=d[re[l]];
return ;
}
int mid=(l+r)>>;
build(f[g].lc,l,mid);
build(f[g].rc,mid+,r);
push_up(g);
}
void add(int g,int l,int r,int c)
{
if(f[g].l>=l && f[g].r<=r)
f[g].flag=true,f[g].cl=f[g].cr=c,f[g].sum=;
else
{
push_down(g);
int mid=(f[g].l+f[g].r)>>;
if(r<=mid) add(f[g].lc,l,r,c);
else if(l>mid) add(f[g].rc,l,r,c);
else add(f[g].lc,l,mid,c),add(f[g].rc,mid+,r,c);
push_up(g);
}
}
void Add(int x,int y,int c)
{
int px=top[x],py=top[y];
while(px!=py)
if(deep[px]>deep[py])
add(rt,dfn[px],dfn[x],c),x=fa[px],px=top[x];
else add(rt,dfn[py],dfn[y],c),y=fa[py],py=top[y];
if(dfn[x]<dfn[y]) add(rt,dfn[x],dfn[y],c);
else add(rt,dfn[y],dfn[x],c);
}
int get(int g,int l,int r)
{
if(f[g].l==L) a=f[g].cl;
if(f[g].r==R) b=f[g].cr;
if(f[g].l>=l && f[g].r<=r)
return f[g].sum;
else
{
push_down(g);
int mid=(f[g].l+f[g].r)>>;
if(r<=mid) return get(f[g].lc,l,r);
else if(l>mid) return get(f[g].rc,l,r);
else
{
int ans=get(f[g].lc,l,mid)+get(f[g].rc,mid+,r);
if(f[f[g].lc].cr==f[f[g].rc].cl) ans--;
return ans;
}
}
}
int Get(int x,int y)
{
int ans=;
int px=top[x],py=top[y];
int l_x=-,l_y=-;
while(px!=py)
if(deep[px]>deep[py])
{
L=dfn[px],R=dfn[x];
ans+=get(rt,dfn[px],dfn[x]);
x=fa[px],px=top[x];
if(b==l_x) ans--;
l_x=a;
}
else
{
L=dfn[py],R=dfn[y];
ans+=get(rt,dfn[py],dfn[y]);
y=fa[py],py=top[y];
if(b==l_y) ans--;
l_y=a;
}
if(dfn[x]>dfn[y]) swap(x,y),swap(l_x,l_y);
L=dfn[x],R=dfn[y];
ans+=get(rt,dfn[x],dfn[y]);
if(a==l_x) ans--;
if(b==l_y) ans--;
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=;i<=n;i++) scanf("%d",&d[i]);
for(int i=;i<n;i++)
{
scanf("%d%d",&u,&v);
link[u].push_back(v);
link[v].push_back(u);
}
dfs1(,);
top[]=,dfs2(,);
build(rt,,cnt);
while(m--)
{
scanf("%s%d%d",s,&u,&v);
if(s[]=='C')
{
scanf("%d",&c);
Add(u,v,c);
}
else if(s[]=='Q')
printf("%d\n",Get(u,v));
}
return ;
}

[SDOI2011]染色(树链剖分)的更多相关文章

  1. BZOJ 2243: [SDOI2011]染色 [树链剖分]

    2243: [SDOI2011]染色 Time Limit: 20 Sec  Memory Limit: 512 MBSubmit: 6651  Solved: 2432[Submit][Status ...

  2. bzoj-2243 2243: [SDOI2011]染色(树链剖分)

    题目链接: 2243: [SDOI2011]染色 Time Limit: 20 Sec  Memory Limit: 512 MBSubmit: 6267  Solved: 2291 Descript ...

  3. 【BZOJ2243】[SDOI2011]染色 树链剖分+线段树

    [BZOJ2243][SDOI2011]染色 Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的 ...

  4. Bzoj 2243: [SDOI2011]染色 树链剖分,LCT,动态树

    2243: [SDOI2011]染色 Time Limit: 20 Sec  Memory Limit: 512 MBSubmit: 5020  Solved: 1872[Submit][Status ...

  5. bzoj2243[SDOI2011]染色 树链剖分+线段树

    2243: [SDOI2011]染色 Time Limit: 20 Sec  Memory Limit: 512 MBSubmit: 9012  Solved: 3375[Submit][Status ...

  6. BZOJ 2243: [SDOI2011]染色 树链剖分 倍增lca 线段树

    2243: [SDOI2011]染色 Time Limit: 20 Sec  Memory Limit: 256 MB 题目连接 http://www.lydsy.com/JudgeOnline/pr ...

  7. BZOJ 2243: [SDOI2011]染色 树链剖分+线段树区间合并

    2243: [SDOI2011]染色 Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的颜色段数 ...

  8. 2243: [SDOI2011]染色(树链剖分+线段树)

    2243: [SDOI2011]染色 Time Limit: 20 Sec  Memory Limit: 512 MBSubmit: 8400  Solved: 3150[Submit][Status ...

  9. Luogu P2486 [SDOI2011]染色(树链剖分+线段树合并)

    Luogu P2486 [SDOI2011]染色 题面 题目描述 输入输出格式 输入格式: 输出格式: 对于每个询问操作,输出一行答案. 输入输出样例 输入样例: 6 5 2 2 1 2 1 1 1 ...

  10. [bzoj 2243]: [SDOI2011]染色 [树链剖分][线段树]

    Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段),如“ ...

随机推荐

  1. SpringBoot2启动流程分析

    首先上一张图,图片来自 SpringBoot启动流程解析 本文基于spring-boot-2.0.4.RELEASE.jar包分析. 程序启动入口 public static void main(St ...

  2. CAS5.3 单点登录/登出/springboot/springmvc

    环境: jdk:1.8 cas server:5.3.14 + tomcat 8.5 cas client:3.5.1 客户端1:springmvc 传统web项目(使用web.xml) 客户端2:s ...

  3. 模型正则化,dropout

    正则化 在模型中加入正则项,防止训练过拟合,使测试集效果提升 Dropout 每次在网络中正向传播时,在每一层随机将一些神经元置零(相当于激活函数置零),一般在全连接层使用,在卷积层一般随机将整个通道 ...

  4. IntelliJ IDEA+springboot+jdbctemplet+easyui+maven+oracle搭建简易开发框架(一)

    前言: 这两天为了巩固easyui的各个控件用法,搭建了一个简易的框架用于开发,大家可以用来参考,如果发现文章中有哪些不正确不合理的地方,也请各位不吝赐教,感激不尽.文章最下面有源码,可以用于参考.整 ...

  5. 给你的 ASP.NET Core 程序插上 Feature Flag 的翅膀

    前言 我们知道,目前大多数应用程序在正式发布到生产环境之前都会经历多个不同的测试环境,通过让应用程序在多个不同的环境中运行来及时发现并解决问题,避免在线上发生不必要的损失.这是对于整个软件的发布流程来 ...

  6. $HDU$ 4336 $Card\ Collector$ 概率$dp$/$Min-Max$容斥

    正解:期望 解题报告: 传送门! 先放下题意,,,已知有总共有$n$张卡片,每次有$p_i$的概率抽到第$i$张卡,求买所有卡的期望次数 $umm$看到期望自然而然想$dp$? 再一看,哇,$n\le ...

  7. 基于Github Pages + docsify,我花了半天就搭建好了个人博客

    目录 前言 一些说明 准备工作 上docsify官网看一看 使用docsify命令生成文档站点 部署到Github上 写在最后 前言 "作为一个真正的码农,不能没有自己的个人博客" ...

  8. PythonI/O进阶学习笔记_9.python的生成器

     content: 1. 什么是生成器 2. 生成器的实现 3. 生成器的应用   一.生成器简介 1.什么是生成器     在 Python 中,使用了 yield 的函数被称为生成器(genera ...

  9. 学了java,我才发现台球还可以这样玩!

    桌球小游戏的尝试 桌球是人们日常生活中都能接触到的一种娱乐活动,随着互联网技术的发展,手机上也有了很多桌球小游戏,让人们随时随地都能打两把.   今天分享一个用java编写的桌球小游戏 代码如下:  ...

  10. 1090 危险品装箱 (25分)C语言

    集装箱运输货物时,我们必须特别小心,不能把不相容的货物装在一只箱子里.比如氧化剂绝对不能跟易燃液体同箱,否则很容易造成爆炸. 本题给定一张不相容物品的清单,需要你检查每一张集装箱货品清单,判断它们是否 ...