题解

不妨采取重链剖分的方式把路径剖成区间,然后用线段树维护,考虑如何合并一个区间

struct Node {
int lf, rg, tot;
}seg[N << 2]; int col[N << 2];
inline Node merge(const Node &lc, const Node &rc) {
if(!lc.tot) return rc;
if(!rc.tot) return lc;
Node ret = (Node){lc.lf, rc.rg, lc.tot + rc.tot};
if(lc.rg == rc.lf) --ret.tot;
return ret;
}

其中$Node$表示线段树中的一个节点,共有三个参数,左端点颜色,右端点颜色以及区间内颜色段数。$col$数组用于下方染色标记。

但是我们要考虑这个区间合并后是否存在相同的颜色其应该只有$1$的贡献却被记了$2$的贡献。这种情况存在当且仅当左区间的右端点颜色与右区间左端点颜色相同。

接着,有关于线段树的其他操作也没有什么好担心的了,接着考虑如何查询。

inline int doit(int x, int y) {
int fx = top[x], fy = top[y];
Node disx = (Node){0, 0, 0}, disy = (Node){0, 0, 0};
while(fx != fy) {
if(dep[fx] >= dep[fy]) disx = merge(query(dfn[fx], dfn[x]), disx), x = fa[fx], fx = top[x];
else disy = merge(query(dfn[fy], dfn[y]), disy), y = fa[fy], fy = top[y];
} if(dfn[x] > dfn[y]) swap(x, y), swap(disx, disy);
swap(disx.lf, disx.rg);
Node ret = merge(merge(disx, query(dfn[x], dfn[y])), disy);
return ret.tot;
}

由于重链剖分跳$top$时,两个端点的路径是独立的,所以不能像普通查询那样直接累加贡献,要分开处理,最后存在一个特殊情况,要将左区间的左右端点反置。(画图即可明白)

代码

#include <cstdio>
#include <algorithm>
using std::swap;
typedef long long ll; const int N = 1e5 + 10;
int n, m, c[N], w[N];
int fa[N], son[N], siz[N], dep[N];
int time, dfn[N], top[N];
int cnt, from[N], to[N << 1], nxt[N << 1];
struct Node {
int lf, rg, tot;
}seg[N << 2]; int col[N << 2]; void addEdge(int u, int v) {
to[++cnt] = v, nxt[cnt] = from[u], from[u] = cnt;
} void dfs(int u) {
dep[u] = dep[fa[u]] + 1, siz[u] = 1;
for(int i = from[u]; i; i = nxt[i]) {
int v = to[i]; if(v == fa[u]) continue;
fa[v] = u, dfs(v), siz[u] += siz[v];
if(siz[v] > siz[son[u]]) son[u] = v;
}
}
void dfs(int u, int t) {
dfn[u] = ++time, top[u] = t, w[time] = c[u];
if(!son[u]) return ; dfs(son[u], t);
for(int i = from[u]; i; i = nxt[i]) {
int v = to[i];
if(v != fa[u] && v != son[u])
dfs(v, v);
}
} inline Node merge(const Node &lc, const Node &rc) {
if(!lc.tot) return rc;
if(!rc.tot) return lc;
Node ret = (Node){lc.lf, rc.rg, lc.tot + rc.tot};
if(lc.rg == rc.lf) --ret.tot;
return ret;
}
inline void pushdown(int o, int lc, int rc) {
if(col[o]) {
seg[lc] = (Node){col[o], col[o], 1};
seg[rc] = (Node){col[o], col[o], 1};
col[lc] = col[rc] = col[o], col[o] = 0;
}
}
void build(int o = 1, int l = 1, int r = n) {
if(l == r) { seg[o] = (Node){w[l], w[l], 1}; return ; }
int mid = (l + r) >> 1, lc = o << 1, rc = lc | 1;
build(lc, l, mid), build(rc, mid + 1, r), seg[o] = merge(seg[lc], seg[rc]);
}
void color(int cl, int cr, int k, int o = 1, int l = 1, int r = n) {
if(l >= cl && r <= cr) {
seg[o] = (Node){k, k, 1}, col[o] = k;
return ;
}
int mid = (l + r) >> 1, lc = o << 1, rc = lc | 1;
pushdown(o, lc, rc);
if(cl <= mid) color(cl, cr, k, lc, l, mid);
if(cr > mid) color(cl, cr, k, rc, mid + 1, r);
seg[o] = merge(seg[lc], seg[rc]);
}
Node query(int ql, int qr, int o = 1, int l = 1, int r = n) {
if(l >= ql && r <= qr) return seg[o];
int mid = (l + r) >> 1, lc = o << 1, rc = lc | 1;
Node ret = (Node){0, 0, 0};
pushdown(o, lc, rc);
if(ql <= mid) ret = query(ql, qr, lc, l, mid);
if(qr > mid) ret = merge(ret, query(ql, qr, rc, mid + 1, r));
return ret;
} inline void upt(int x, int y, int k) {
int fx = top[x], fy = top[y];
while(fx != fy) {
if(dep[fx] >= dep[fy]) color(dfn[fx], dfn[x], k), x = fa[fx], fx = top[x];
else color(dfn[fy], dfn[y], k), y = fa[fy], fy = top[y];
} if(dfn[x] > dfn[y]) swap(x, y);
color(dfn[x], dfn[y], k);
}
inline int doit(int x, int y) {
int fx = top[x], fy = top[y];
Node disx = (Node){0, 0, 0}, disy = (Node){0, 0, 0};
while(fx != fy) {
if(dep[fx] >= dep[fy]) disx = merge(query(dfn[fx], dfn[x]), disx), x = fa[fx], fx = top[x];
else disy = merge(query(dfn[fy], dfn[y]), disy), y = fa[fy], fy = top[y];
} if(dfn[x] > dfn[y]) swap(x, y), swap(disx, disy);
swap(disx.lf, disx.rg);
Node ret = merge(merge(disx, query(dfn[x], dfn[y])), disy);
return ret.tot;
} int main () {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i) scanf("%d", c + i);
for (int i = 1, u, v; i < n; ++i) {
scanf("%d%d", &u, &v);
addEdge(u, v), addEdge(v, u);
}
dfs(1), dfs(1, 1), build();
char opt; int a, b, c;
while(m--) {
scanf("\n%c%d%d", &opt, &a, &b);
if(opt == 'C') {
scanf("%d", &c);
upt(a, b, c);
} else printf("%d\n", doit(a, b));
}
return 0;
}

Luogu P2486 染色(树链剖分+线段树)的更多相关文章

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

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

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

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

  3. B20J_2243_[SDOI2011]染色_树链剖分+线段树

    B20J_2243_[SDOI2011]染色_树链剖分+线段树 一下午净调这题了,争取晚上多做几道. 题意: 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成 ...

  4. 2243: [SDOI2011]染色 树链剖分+线段树染色

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

  5. BZOJ2243 [SDOI2011]染色(树链剖分+线段树合并)

    题目链接 BZOJ2243 树链剖分 $+$ 线段树 线段树每个节点维护$lc$, $rc$, $s$ $lc$代表该区间的最左端的颜色,$rc$代表该区间的最右端的颜色 $s$代表该区间的所有连续颜 ...

  6. BZOJ2243 (树链剖分+线段树)

    Problem 染色(BZOJ2243) 题目大意 给定一颗树,每个节点上有一种颜色. 要求支持两种操作: 操作1:将a->b上所有点染成一种颜色. 操作2:询问a->b上的颜色段数量. ...

  7. 【bzoj1959】[Ahoi2005]LANE 航线规划 树链剖分+线段树

    题目描述 对Samuel星球的探险已经取得了非常巨大的成就,于是科学家们将目光投向了Samuel星球所在的星系——一个巨大的由千百万星球构成的Samuel星系. 星际空间站的Samuel II巨型计算 ...

  8. 洛谷P3313 [SDOI2014]旅行 题解 树链剖分+线段树动态开点

    题目链接:https://www.luogu.org/problem/P3313 这道题目就是树链剖分+线段树动态开点. 然后做这道题目之前我们先来看一道不考虑树链剖分之后完全相同的线段树动态开点的题 ...

  9. 【BZOJ-2325】道馆之战 树链剖分 + 线段树

    2325: [ZJOI2011]道馆之战 Time Limit: 40 Sec  Memory Limit: 256 MBSubmit: 1153  Solved: 421[Submit][Statu ...

  10. POJ3237 (树链剖分+线段树)

    Problem Tree (POJ3237) 题目大意 给定一颗树,有边权. 要求支持三种操作: 操作一:更改某条边的权值. 操作二:将某条路径上的边权取反. 操作三:询问某条路径上的最大权值. 解题 ...

随机推荐

  1. LightOJ 1284 - Lights inside 3D Grid 概率/期望/二项式定理

    题意:给你一个长宽高为x,y,z的长方体,里面每个格子放了灯,再给你k次选取任意长方体形状的区块,对其内所有灯开或关操作,初始为关,问亮灯数量的期望值. 题解:首先考虑选取区块的概率,使某个灯在被选取 ...

  2. Jenkenis报错:该jenkins实例似乎已离线[转]

    解决方法: 安装插件那个页面,就是提示你offline的那个页面,不要动.然后打开一个新的tab,输入网址http://localhost:8080/pluginManager/advanced. 这 ...

  3. Bzoj2832 / Bzoj3874 宅男小C

    Time Limit: 10 Sec  Memory Limit: 256 MBSubmit: 124  Solved: 26 Description 众所周知,小C是个宅男,所以他的每天的食物要靠外 ...

  4. 作为一名前端开发工程师,你必须掌握的WEB模板引擎:Handlebars

    作为一名前端开发工程师,你必须掌握的WEB模板引擎:Handlebars 一.为什么需要使用模板引擎? 关于为什么要使用模板引擎,按照我常对学生说的一句话就是:不用重复造轮子..   简单来说,模板最 ...

  5. 安装magento配置完数据库后出现: PHP Extensions "0" must be loaded. 解决方案

    打开magento目录下的该文件: app/code/core/Mage/Install/etc/config.xml 将下面该段修改: <mysql4> <type>pdo_ ...

  6. 关于this问题

    对于关键字this,其实很好理解,谁调用我就指向谁.下面举个例子说明: 其实这也是在学习闭包中的一个案例: var name = "The window"; var obj = { ...

  7. mysql 复制表结构 / 从结果中导入数据到新表

    这只会复制结构: mysql> create table a like mysql1; Query OK, 0 rows affected (0.03 sec) mysql> desc a ...

  8. Ubuntu 14.04开启ssh服务

    sudo apt-get install openssh-server sudo apt-get install openssh-client sudo service ssh restart

  9. 基于ARM 构架(带MMU)的copy_from_user与copy_to_user详细分析

      [转自:http://blog.chinaunix.net/uid-20543672-id-3195249.html]   在学习Linux内核驱动的时候,一开始就会碰到copy_from_use ...

  10. C语言restrict限定符

    restrict是c99标准引入的,它只可以用于限定和约束指针,并表明指针是访问一个数据对象的唯一且初始的方式.即它告诉编译器,所有修改该指针所指向内存中内容的操作都必须通过该指针来修改,而不能通过其 ...