P2486 [SDOI2011]染色
树链剖分
用区间修改线段树维护
对于颜色段的计算:sum[o]=sum[lc]+sum[rc]
因为可能重复计算,即左子树的右端点和右子树的左端点可能颜色相同
多开2个数组lx,rx记录左/右端点的颜色,重复的话 sum[o]- - 即可
(我的错误模板坑了我2h)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cctype>
using namespace std;
template <typename T> inline T min(T &a,T &b) {return a<b ?a:b;}
template <typename T> inline T max(T &a,T &b) {return a>b ?a:b;}
template <typename T> inline void read(T &x){
char c=getchar(); x=; bool f=;
while(!isdigit(c)) f= !f||c=='-' ? :,c=getchar();
while(isdigit(c)) x=(x<<)+(x<<)+(c^),c=getchar();
x= f ? x:-x;
}
template <typename T> inline void output(T x){
if(!x) {putchar(); return ;}
if(x<) putchar('-'),x=-x;
int wt[],l=;
while(x) wt[++l]=x%,x/=;
while(l) putchar(wt[l--]+);
}
typedef int arr[];
int n,m,cnt,tot;
arr d,fa,siz,bgs,tp,tmp,val,id,hd,ed;
int nxt[],poi[];
int lx[],rx[],sum[],tag[];
inline void add_(int x,int y){
nxt[ed[x]]=++cnt; hd[x]= hd[x] ? hd[x]:cnt;
ed[x]=cnt; poi[cnt]=y;
}
inline void pushdown(int o){
if(tag[o]==-) return ;
int lc=o<<,rc=o<<|;
sum[lc]=sum[rc]=;
lx[lc]=rx[lc]=lx[rc]=rx[rc]=tag[o];
tag[lc]=tag[rc]=tag[o];
tag[o]=-;
}
inline void maintain(int o){
int lc=o<<,rc=o<<|;
lx[o]=lx[lc],rx[o]=rx[rc],sum[o]=sum[lc]+sum[rc];
if(rx[lc]==lx[rc]) --sum[o]; //重复减掉
}
inline void build(int o,int l,int r){
tag[o]=-;
if(l==r) {tag[o]=lx[o]=rx[o]=tmp[l]; sum[o]=; return ;}
int lc=o<<,rc=o<<|,mid=l+((r-l)>>);
build(lc,l,mid); build(rc,mid+,r);
maintain(o);
}
inline void modify(int o,int l,int r,int x1,int x2,int v){
if(x1<=l&&r<=x2){
lx[o]=rx[o]=tag[o]=v; sum[o]=;
return ;
}pushdown(o);
int lc=o<<,rc=o<<|,mid=l+((r-l)>>);
if(x1<=mid) modify(lc,l,mid,x1,x2,v);
if(x2>mid) modify(rc,mid+,r,x1,x2,v);
maintain(o);
}
inline int query(int o,int l,int r,int x1,int x2){
if(x1<=l&&r<=x2) return sum[o];
pushdown(o); int res=;
int lc=o<<,rc=o<<|,mid=l+((r-l)>>);
if(x1<=mid) res+=query(lc,l,mid,x1,x2);
if(x2>mid){
res+=query(rc,mid+,r,x1,x2);
if(x1<=mid&&rx[lc]==lx[rc]) --res;
}return res;
}
inline void dfs1(int x,int _fa){
d[x]=d[_fa]+,fa[x]=_fa,siz[x]=;
for(int i=hd[x];i;i=nxt[i])
if(poi[i]!=_fa){
dfs1(poi[i],x);
siz[x]+=siz[poi[i]];
if(siz[bgs[x]]<siz[poi[i]]) bgs[x]=poi[i];
}
}
inline void dfs2(int x,int _top){
id[x]=++tot,tmp[tot]=val[x],tp[x]=_top;
if(siz[x]==) return;
dfs2(bgs[x],_top);
for(int i=hd[x];i;i=nxt[i])
if(poi[i]!=fa[x]&&poi[i]!=bgs[x])
dfs2(poi[i],poi[i]);
}
inline int findcol(int o,int l,int r,int x){
if(l==r) return tag[o];
pushdown(o);
int lc=o<<,rc=o<<|,mid=l+((r-l)>>);
if(x<=mid) return findcol(lc,l,mid,x);
else return findcol(rc,mid+,r,x);
}
inline void change(int x,int y,int v){
while(tp[x]!=tp[y]){ //™模板原来这里多套了个d数组上去,变成50pts
if(d[tp[x]]<d[tp[y]]) swap(x,y);
modify(,,n,id[tp[x]],id[x],v);
x=fa[tp[x]];
}if(d[x]>d[y]) swap(x,y);
modify(,,n,id[x],id[y],v);
}
inline int ask(int x,int y){
int res=,r1,r2;
while(tp[x]!=tp[y]){
if(d[tp[x]]<d[tp[y]]) swap(x,y);
res+=query(,,n,id[tp[x]],id[x]);
r1=findcol(,,n,id[tp[x]]);
r2=findcol(,,n,id[fa[tp[x]]]);
if(r1==r2) --res;
x=fa[tp[x]];
}if(d[x]>d[y]) swap(x,y);
res+=query(,,n,id[x],id[y]);
return max(res,);
}
int main(){
read(n); read(m); int q1,q2,q3; char opt[];
for(int i=;i<=n;++i) read(val[i]);
for(int i=;i<n;++i) read(q1),read(q2),add_(q1,q2),add_(q2,q1);
dfs1(,); dfs2(,); build(,,n);
for(int i=;i<=m;++i){
scanf("%s",opt); read(q1),read(q2);
if(opt[]=='Q') output(ask(q1,q2)),putchar('\n');
else read(q3),change(q1,q2,q3);
}return ;
}
P2486 [SDOI2011]染色的更多相关文章
- Luogu P2486 [SDOI2011]染色(树链剖分+线段树合并)
Luogu P2486 [SDOI2011]染色 题面 题目描述 输入输出格式 输入格式: 输出格式: 对于每个询问操作,输出一行答案. 输入输出样例 输入样例: 6 5 2 2 1 2 1 1 1 ...
- 洛谷 P2486 [SDOI2011]染色 树链剖分
目录 题面 题目链接 题目描述 输入输出格式 输入格式 输出格式 输入输出样例 输入样例: 输出样例: 说明 思路 PushDown与Update Q AC代码 总结与拓展 题面 题目链接 P2486 ...
- 洛谷 P2486 [SDOI2011]染色/bzoj 2243: [SDOI2011]染色 解题报告
[SDOI2011]染色 题目描述 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同 ...
- luogu P2486 [SDOI2011]染色
树剖做法: 就是两个dfs+一个线段树 难度的取决基本==线段树的维护难度 所以对有点线段树基础的,树剖也不难做吧 这里操作有二 一:两点间路径染色 线段树的区间赋值操作 二:查询路径段的个数 考虑线 ...
- P2486 [SDOI2011]染色(树剖)区间覆盖+区间的连续段
https://www.luogu.org/problemnew/show/P2486 值的一看https://www.cnblogs.com/Tony-Double-Sky/p/9283262.ht ...
- 洛谷 P2486 [SDOI2011]染色 LCT
Code: #include <cstdio> //SDOI2010 染色 #include <algorithm> #include <cstring> #inc ...
- P2486 [SDOI2011]染色 维护区间块数 树链剖分
https://www.luogu.org/problemnew/show/P2486 题意 对一个树上维护两种操作,一种是把x到y间的点都染成c色,另一种是求x到y间的点有多少个颜色块,比如11 ...
- luogu题解P2486[SDOI2011]染色--树链剖分+trick
题目链接 https://www.luogu.org/problemnew/show/P2486 分析 看上去又是一道强行把序列上问题搬运到树上的裸题,然而分析之后发现并不然... 首先我们考虑如何在 ...
- 洛谷P2486 [SDOI2011]染色 题解 树链剖分+线段树
题目链接:https://www.luogu.org/problem/P2486 首先这是一道树链剖分+线段树的题. 线段树部分和 codedecision P1112 区间连续段 一模一样,所以我们 ...
随机推荐
- uva 10983 Buy one, get the rest free 二分判定层次图
二分枚举租用飞机的最大花费,然后用小于等于最大花费的边构建层次图(依据时间) 构图思路: 利用二元组(x,y)表示 x天y城市 1. e天有飞机从a城市飞到b城市,能够承载x人,则添加单向边 ( ...
- Linux下pip使用国内源
pip国内的一些镜像 阿里云 http://mirrors.aliyun.com/pypi/simple/ 中国科技大学 https://pypi.mirrors.ustc.edu.cn/si ...
- CSS垂直导航栏
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8&quo ...
- Oracle核心技术之 SQL TRACE
1.SQL TRACE说明: 参数类型 布尔型 缺省值 false 参数类别 动态 取值范围 True|false 2.类型 1)sql trace参数:alter system改变对全局进程影响,如 ...
- Oracle备份恢复之逻辑备份
exp 交互模式:导出scott用户下的emp表. [oracle@localhost ~]$ exp Export: Release 10.2.0.1.0 - Production on Thu N ...
- python面向对象高级:@property
@property 把方法『变成』了属性,广泛应用在类的定义中,可以让调用者写出简短的代码,同时保证对参数进行必要的检查,这样,程序运行时就减少了出错的可能性. 最大的作用就是既能检查参数,又可以用类 ...
- Django实现微电影网站
主要实现功能: 注册,登陆,播放视频,评论视频,xadmin实现后台提供视频管理,权限管理等等 首页(搜索,筛选) 搜索页面 播放页面 个人中心 后台 源码下载 环境配置:python2.7 pip ...
- 洛谷 P4201 设计路线 [NOI2008] 树形dp
正解:树形dp 解题报告: 大概是第一道NOI的题目?有点激动嘻嘻 然后先放个传送门 先大概港下这题的题意是啥qwq 大概就是给一棵树,然后可以选若干条链把链上的所有边的边权变成0,但是这些链不能有交 ...
- Redis添加历史浏览记录
参考资料 http://redisdoc.com/index.html http://redis-py.readthedocs.io/en/latest/#indices-and-tables 1.什 ...
- Storyboard Reference
在某些情况下.你开发的应用可能包括有各种各样的复杂界面,假设你使用Storyboard来管理这些界面,通常你会将这些界面依照对应的逻辑分成很多子模块,放在不同的storyboard里,比方注冊相关的放 ...