传送门

就是个模板啦

记录每一个点的左端点颜色和右端点颜色和当前端点颜色段数。

合并时如果左孩子右端点和右孩子左端点不同就 ans--

在重链上跳的时候别忘记统计一下

——代码

#include <cstdio>
#include <cstring>
#include <iostream>
#define N 400001
#define ls now << 1
#define rs now << 1 | 1
#define swap(x, y) ((x) ^= (y) ^= (x) ^= (y))
#define pushup(now) Lcol[now] = Lcol[ls], Rcol[now] = Rcol[rs], col[now] = col[ls] + col[rs] - (Rcol[ls] == Lcol[rs]) int n, m, cnt, tim;
int head[N], to[N << 1], next[N << 1];
int a[N], f[N], deep[N], size[N], son[N], tid[N], top[N], rank[N], L[N], R[N], Lcol[N], Rcol[N], col[N], laz[N]; inline int read()
{
int x = 0, f = 1;
char ch = getchar();
for(; !isdigit(ch); ch = getchar()) if(ch == '-') f = -1;
for(; isdigit(ch); ch = getchar()) x = (x << 1) + (x << 3) + ch - '0';
return x * f;
} inline void add(int x, int y)
{
to[cnt] = y;
next[cnt] = head[x];
head[x] = cnt++;
} inline void dfs1(int u)
{
int i, v;
size[u] = 1;
deep[u] = deep[f[u]] + 1;
for(i = head[u]; i ^ -1; i = next[i])
{
v = to[i];
if(v ^ f[u])
{
f[v] = u;
dfs1(v);
size[u] += size[v];
if(son[u] == -1 || size[son[u]] < size[v]) son[u] = v;
}
}
} inline void dfs2(int u, int t)
{
int i, v;
top[u] = t;
tid[u] = ++tim;
rank[tim] = u;
if(son[u] == -1) return;
dfs2(son[u], t);
for(i = head[u]; i ^ -1; i = next[i])
{
v = to[i];
if(v ^ f[u] && v ^ son[u]) dfs2(v, v);
}
} inline void build(int now, int l, int r)
{
L[now] = l;
R[now] = r;
if(l == r)
{
col[now] = 1;
Lcol[now] = Rcol[now] = a[rank[l]];
return;
}
int mid = (l + r) >> 1;
build(ls, l, mid);
build(rs, mid + 1, r);
pushup(now);
} inline void pushdown(int now)
{
if(laz[now])
{
col[ls] = col[rs] = 1;
laz[ls] = laz[rs] = Lcol[ls] = Lcol[rs] = Rcol[ls] = Rcol[rs] = laz[now];
laz[now] = 0;
}
} inline void update(int now, int l, int r, int z)
{
if(l <= L[now] && R[now] <= r)
{
col[now] = 1;
laz[now] = Lcol[now] = Rcol[now] = z;
return;
}
pushdown(now);
int mid = (L[now] + R[now]) >> 1;
if(l <= mid) update(ls, l, r, z);
if(mid < r) update(rs, l, r, z);
pushup(now);
} inline void q_update(int u, int v, int z)
{
while(top[u] ^ top[v])
{
if(deep[top[u]] < deep[top[v]]) swap(u, v);
update(1, tid[top[u]], tid[u], z);
u = f[top[u]];
}
if(deep[u] > deep[v]) swap(u, v);
update(1, tid[u], tid[v], z);
} inline int q_color(int now, int x)
{
if(L[now] == R[now]) return Lcol[now];
pushdown(now);
int mid = (L[now] + R[now]) >> 1;
if(x <= mid) return q_color(ls, x);
else return q_color(rs, x);
} inline int query(int now, int l, int r)
{
if(l <= L[now] && R[now] <= r) return col[now];
pushdown(now);
int mid = (L[now] + R[now]) >> 1, ans = 0;
if(l <= mid) ans += query(ls, l, r);
if(mid < r) ans += query(rs, l, r);
if(l <= mid && mid < r) ans -= (Rcol[ls] == Lcol[rs]);
return ans;
} inline int q_query(int u, int v)
{
int ans = 0;
while(top[u] ^ top[v])
{
if(deep[top[u]] < deep[top[v]]) swap(u, v);
ans += query(1, tid[top[u]], tid[u]);
ans -= (q_color(1, tid[f[top[u]]]) == q_color(1, tid[top[u]]));
u = f[top[u]];
}
if(deep[u] > deep[v]) swap(u, v);
ans += query(1, tid[u], tid[v]);
return ans;
} int main()
{
int i, x, y, z;
char s[10];
n = read();
m = read();
memset(son, -1, sizeof(son));
memset(head, -1, sizeof(head));
for(i = 1; i <= n; i++) a[i] = read();
for(i = 1; i < n; i++)
{
x = read();
y = read();
add(x, y);
add(y, x);
}
dfs1(1);
dfs2(1, 1);
build(1, 1, n);
for(i = 1; i <= m; i++)
{
scanf("%s", s);
x = read();
y = read();
if(s[0] == 'C')
{
z = read();
q_update(x, y, z);
}
else printf("%d\n", q_query(x, y));
}
return 0;
}

  

[luoguP2486] [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. P3482 [POI2009]SLO-Elephants

    传送门 首先,交换关系肯定是构成一个环的时候最优 如果这个环是自环,不用交换了 如果环的大小为2,直接交换便是 否则的话,我们可以用环里最小的点最为交换媒介,然后去和其他交换直到到达正确的位置,那么环 ...

  2. Final关键字解析

    final 在 Java 中是一个保留的关键字,可以声明变量.方法.类. 什么是final变量 / 类 / 方法? 任何变量前被 final 修饰就是 final 变量,定义的类前被 final 修饰 ...

  3. Linux学习笔记之Linux shell脚本运行出现问题:bash: ./test: bin/sh: bad interpreter: No such file or directory

    问题: 在Linux系统中使用“vi test.sh”命令创建.sh文件,保存文件(:wq)并赋予权限(chmod +x test.sh)后,执行(./test.sh),出现问题:“bash: ./t ...

  4. Linux环境下修改MySQL数据库对表名大小写不敏感

    Linux系统中MySQL对数据库名称和表名是大小写敏感的,这就导致了一些麻烦,虽然已经建立了表和数据,但因为大小写导致无法找到表. MySQL数据库对表名大小写不敏感的设置方法如下: 1.查看MyS ...

  5. RHEL5.6更新yum源

    RHEL5.6更新yum源记录,2017年2月20日 root用户切换目录至:/etc/yum.repos.d/ [root@localhost yum.repos.d]# pwd /etc/yum. ...

  6. js作用域、异步——学习笔记

    所有的 for if switch while do 等等,都属于块级作用域,里面声明的对象,外面也能访问.但function 函数里的作用域,在函数外是访问不到的. 但函数作用域里面可以访问外面. ...

  7. 类支付宝密码输入框NumberEditText(简单粗暴的定制方式)

    因为项目需要,设计了一个下图样的验证码输入框(ps:个人认为还不如直接一个EditText,用户友好度可能更好,何况这页面99.9%的用户不会使用,但是没办法,别人才是专业的设计师). 其实界面很简单 ...

  8. acm练习-day1_1

    “麻雀”lengdan用随机数生成了后台数据,但是笨笨的他被妹纸的问题给难住了... 已知lengdan生成了N(1=<N<=10005)个随机整数,妹子对这些数可能有以下几种操作或询问: ...

  9. 【Python-2.7】删除空格

    有时我们在编程过程中,需要去除字符串两边的空格,可以用如下函数解决问题: rstrip():去除字符串右边的空格: lstrip():去除字符串左边的空格: strip():去除字符串两边的空格. 示 ...

  10. parsley之验证属性设置

    parsley.js添加表单验证功能,直接在html元素中添加对应属性: Name API Description Required #2.0必填 required HTML5 data-parsle ...