【BZOJ2243】【SDOI2011】染色
题意见试题传送门
解题思路:显然是题树剖题。
考虑用线段树维护区间端点颜色与颜色数,这样就可以方便的合并,注意查询的时候对端点的特殊处理即可。
时间效率最高为\( O (m \log^{2} n) \).(BZOJ 上 4072ms)
#include <stdio.h>
#define MN 100005
#define Mn (1<<17)
#define ls(x) (x<<1)
#define rs(x) (x<<1|1)
#define mid (l+r>>1)
int lc[Mn<<],rc[Mn<<],sum[Mn<<],mark[Mn<<];
int n,q,col[MN],siz[MN],fa[MN],son[MN],dep[MN],top[MN],pos[MN],head[MN],cnt,dfsn,rev[MN];
int to[MN<<],nxt[MN<<];
inline int in(){
int x=;bool f=; char ch=getchar();
while(ch<''||ch>'') f=ch=='-',ch=getchar();
while(ch>=''&&ch<='') x=(x<<)+(x<<)+ch-'',ch=getchar();
return f?-x:x;
}
inline void swp(int &a,int &b){a^=b^=a^=b;}
inline void ins(int x,int y){to[++cnt]=y,nxt[cnt]=head[x],head[x]=cnt;}
inline void dfs1(int u,int f,int d){
fa[u]=f,dep[u]=d,siz[u]=;
for (register int i=head[u]; i; i=nxt[i])
if (to[i]!=f){
dfs1(to[i],u,d+);siz[u]+=siz[to[i]];
if (siz[to[i]]>siz[son[u]]) son[u]=to[i];
}
}
inline void dfs2(int u,int tp){
top[u]=tp;pos[u]=(++dfsn);rev[dfsn]=u;if (son[u]) dfs2(son[u],tp);
for (register int i=head[u]; i; i=nxt[i])
if (to[i]!=fa[u]&&to[i]!=son[u]) dfs2(to[i],to[i]);
}
inline void pushdown(int k){
mark[ls(k)]=mark[rs(k)]=lc[ls(k)]=rc[ls(k)]=lc[rs(k)]=rc[rs(k)]=mark[k];
sum[ls(k)]=sum[rs(k)]=;mark[k]=;
}
inline void combine(int k){
sum[k]=rc[ls(k)]==lc[rs(k)]?sum[ls(k)]+sum[rs(k)]-:sum[ls(k)]+sum[rs(k)];
lc[k]=lc[ls(k)],rc[k]=rc[rs(k)];
}
inline void build(int k,int l,int r){
mark[k]=;
if (l==r){sum[k]=,lc[k]=rc[k]=col[rev[l]];return;}
build(ls(k),l,mid);build(rs(k),mid+,r);combine(k);
}
inline void A(int l,int r,int a,int b,int k,int col){
if (a<=l&&r<=b){mark[k]=lc[k]=rc[k]=col;sum[k]=;return;}
if (mark[k]) pushdown(k);if (a<=mid) A(l,mid,a,b,ls(k),col);
if (b>mid) A(mid+,r,a,b,rs(k),col);combine(k);
}
inline int Qs(int l,int r,int a,int b,int k){
if (l==a&&r==b) return sum[k];if (mark[k]) pushdown(k);
if (b<=mid) return Qs(l,mid,a,b,ls(k));
if (a>mid) return Qs(mid+,r,a,b,rs(k));
return Qs(l,mid,a,mid,ls(k))+Qs(mid+,r,mid+,b,rs(k))-(rc[ls(k)]==lc[rs(k)]);
}
inline int Qc(int l,int r,int x,int k){
if (l==r) return lc[k];if (mark[k]) pushdown(k);
if (x<=mid) return Qc(l,mid,x,ls(k));return Qc(mid+,r,x,rs(k));
}
inline void update(int x,int y,int cl){
while (top[x]!=top[y]){
if (dep[top[x]]<dep[top[y]]) swp(x,y);
A(,n,pos[top[x]],pos[x],,cl);x=fa[top[x]];
}if (dep[x]>dep[y]) swp(x,y);A(,n,pos[x],pos[y],,cl);
}
inline int query(int x,int y){
register int res=;
while(top[x]!=top[y]){
if (dep[top[x]]<dep[top[y]]) swp(x,y);
res+=Qs(,n,pos[top[x]],pos[x],)-(Qc(,n,pos[top[x]],)==Qc(,n,pos[fa[top[x]]],));
x=fa[top[x]];
}if (dep[x]>dep[y]) swp(x,y);res+=Qs(,n,pos[x],pos[y],);return res;
}
void init(){
n=in(),q=in();for (int i=; i<=n; ++i) col[i]=in();
for (register int i=; i<n; ++i){
register int x=in(),y=in();
ins(x,y); ins(y,x);
}dfs1(,,);dfs2(,);build(,,n);
}
void solve(){
while(q--){
register char ch=getchar();while (ch!='C'&&ch!='Q') ch=getchar();
register int x=in(),y=in();
if (ch=='C') update(x,y,in());
else printf("%d\n",query(x,y));
}
}
int main(){init(); solve(); return ;}
【BZOJ2243】【SDOI2011】染色的更多相关文章
- BZOJ2243 SDOI2011 染色 【树链剖分】
BZOJ2243 SDOI2011 染色 Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的颜色 ...
- bzoj2243[SDOI2011]染色 树链剖分+线段树
2243: [SDOI2011]染色 Time Limit: 20 Sec Memory Limit: 512 MBSubmit: 9012 Solved: 3375[Submit][Status ...
- [bzoj2243][SDOI2011]染色
Description 给定一棵有$n$个节点的无根树和$m$个操作,操作有$2$类: 1.将节点$a$到节点$b$路径上所有点都染成颜色$c$; 2.询问节点$a$到节点$b$路径上的颜色段数量(连 ...
- BZOJ2243[SDOI2011]染色——树链剖分+线段树
题目描述 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段), 如“112221 ...
- [BZOJ2243][SDOI2011]染色 解题报告|树链剖分
Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段),如“ ...
- BZOJ2243 [SDOI2011]染色(树链剖分+线段树合并)
题目链接 BZOJ2243 树链剖分 $+$ 线段树 线段树每个节点维护$lc$, $rc$, $s$ $lc$代表该区间的最左端的颜色,$rc$代表该区间的最右端的颜色 $s$代表该区间的所有连续颜 ...
- BZOJ2243: [SDOI2011]染色(树链剖分/LCT)
Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段), 如 ...
- bzoj2243: [SDOI2011]染色--线段树+树链剖分
此题代码量较大..但是打起来很爽 原本不用lca做一直wa不知道为什么.. 后来改lca重打了一遍= =结果一遍就AC了orz 题目比较裸,也挺容易打,主要是因为思路可以比较清晰 另:加读入优化比没加 ...
- BZOJ2243——[SDOI2011]染色
1.题目大意:给个树,然后树上每个点都有颜色,然后会有路径的修改,有个询问,询问一条路径上的颜色分成了几段 2.分析:首先这个修改是树剖可以做的,对吧,但是这个分成了几段怎么搞呢,我们的树剖的不是要建 ...
- bzoj2243 sdoi2011 染色 paint
明明是裸树剖 竟然调了这么久好蛋疼 大概是自己比较水的原因吧 顺便+fastio来gangbang #include<iostream> #include<cstring> # ...
随机推荐
- Alpha冲刺No.1
冲刺Day1 一.站立式会议计划 全体成员先安装好Android Studio,mysql,以及navicat for MySQL 将上述软件调试至可运行状态 自主把玩安卓虚拟机,mysql 通过一些 ...
- python控制流 If-else
控制流 If-else 我们处理现实生活中的问题时会做出决定,就像决定买哪种相机或者怎样更好的打篮球.同样我们写计算机程序的时候也要做相同的事情.我们通过 if-else 语句来做决定,我们使 ...
- hashlib 加密
import hashlib def md5(args): hash = hashlib.md5(bytes('aaadf',encoding='utf-8')) hash.update(bytes( ...
- java从网络中下载图片到本地
public class imageDownload { public static void main(String[] args) { String url = "http://loca ...
- 第四十三条:返回零长度的数组或者集合,而不是null
如果一个方法的返回值类型是集合或者数组 ,如果在方法内部需要返回的集合或者数组是零长度的,也就是没有实际对象在里面, 我们也应该放回一个零长度的数组或者集合,而不是返回null.如果返回了null,客 ...
- R语言基础2
----------------------------------R语言学习与科研应用,科研作图,数据统计挖掘分析,群:719954246-------------------------- 通常, ...
- 09-移动端开发教程-Sass入门
1. 引言 CSS3之前的CSS都大都是枚举属性样式,而编程语言强大的变量.函数.循环.分支等功能基本都不能在CSS中使用,让CSS的编程黯淡无光,Sass就是一种增强CSS编程的扩展语言(CSS4也 ...
- 11-TypeScript中的名称空间
在后端开发语言中,比如C#中,可以将不同源代码文件中的代码通过名称空间组合到一起.一般一个类定义在一个源代码文件中,在功能上属于一个上下文的源代码文件通过名称空间进行组织. 在TypeScript中, ...
- 根据抽象工厂实现的DBHelpers类
public abstract class DBHelper { public static SqlConnection conn = new SqlConnection("server=l ...
- IDEA插件和快捷设置
前言 IDEA全名Intellij IDEA,是Java开发的集成环境,它有两个版本,专业版(Ultimate)和社区版(Community),专业版需要注册,而社区版不用注册,同时需要注意的是社区版 ...