题意:一棵树,点有$0,1,2$三种颜色,支持路径修改颜色和查询点所在同色连通块的大小

lcm太可怕了,于是去问了sk,得到一个优质做法

考虑lct维护子树信息,$vs_{x,i}$为$x$的虚儿子中,以颜色为$i$的节点为根的同色连通块大小之和,$s_{x,i}$表示splay上$x$的子树$vs_{x,i}$之和,切换虚实时更新$vs$,splay上pushup时更新$s$即可

如果每时每刻都保持同一棵splay中点的颜色都相同,那么询问时只需模仿access的过程,不停往上拼接同色splay,最后得到的splay的根节点$x$的$s_{x,c_x}+siz_x$就是答案,我们把这种access称为按颜色access

现在考虑修改,先求lca,把修改拆成两个祖先后代链,假设这条祖先后代链为$y\rightarrow x$,$y$是$x$的祖先

先对$fa_y$按颜色access,再对$x\rightarrow y$无条件access,对得到的splay打标记即可

这棵splay可能会作为某个点的虚儿子,看起来要一直往上更新,实际上最多更新往上的两棵splay即可

设往上的三棵splay为$T_1,T_2,T_3$,因为$T_1$是按颜色access得到的,所以$c_{T_1}\neq c_{T_2}$

首先$T_1$显然需要更新,然后因为$T_2$需要用到$T_1$的信息,所以当$y\rightarrow x$这条链在修改前或修改后的颜色$=c_{T_1}$时,$T_1$的$s_{x,c_{T_1}}$会变化,进而影响$T_2$的$s_{x,c_{T_1}}$

幸运地,$T_3$只需要用到$T_2$的$s_{x,c_{T_2}}$信息,又因为$c_{T_1}\ne c_{T_2}$,所以从$T_3$开始往上的那些splay都无需更新

最后是无条件access所引发的一些小问题,在按颜色access时,我们可以快速而准确地更新一个节点的$vs$,但无条件access时,splay中可能含有不同颜色的点,这时不能直接用$s_{x,c_x}+siz_x$来计算$x$对父亲的$vs$的贡献

解决方法很简单:在拼接$x$和$y$之前先算出$x$对父亲的旧贡献,减掉即可,又因为一个节点原来的实儿子的splay子树中都是同颜色的点,这部分的贡献可以直接按原来的方法算

于是整道题就做完了,这个题还是挺好的==

再次orzskdtz两位人形自走dspedia

#include<stdio.h>
#include<algorithm>
using namespace std;
int n;
namespace t{
	int h[100010],nex[100010],to[100010],M;
	void add(int a,int b){
		M++;
		to[M]=b;
		nex[M]=h[a];
		h[a]=M;
	}
	int fa[100010][17],dep[100010],siz[100010];
	void dfs(int x){
		dep[x]=dep[fa[x][0]]+1;
		siz[x]=1;
		for(int i=h[x];i;i=nex[i]){
			if(to[i]!=fa[x][0]){
				dfs(to[i]);
				siz[x]+=siz[to[i]];
			}
		}
	}
	void work(){
		int i,j;
		for(i=2;i<=n;i++){
			scanf("%d",fa[i]);
			add(fa[i][0],i);
		}
		dfs(1);
		for(j=1;j<17;j++){
			for(i=1;i<=n;i++)fa[i][j]=fa[fa[i][j-1]][j-1];
		}
	}
	int lca(int x,int y){
		int i;
		if(dep[x]<dep[y])swap(x,y);
		for(i=16;i>=0;i--){
			if(dep[fa[x][i]]>=dep[y])x=fa[x][i];
		}
		if(x==y)return x;
		for(i=16;i>=0;i--){
			if(fa[x][i]!=fa[y][i]){
				x=fa[x][i];
				y=fa[y][i];
			}
		}
		return fa[x][0];
	}
}
namespace l{
	int ch[100010][2],fa[100010],r[100010],siz[100010],s[100010][3],vs[100010][3],d[100010],c[100010];
	#define ls ch[x][0]
	#define rs ch[x][1]
	void pushup(int x){
		for(int i=0;i<3;i++)s[x][i]=s[ls][i]+s[rs][i]+vs[x][i];
		r[x]=rs?r[rs]:x;
		siz[x]=siz[ls]+siz[rs]+1;
	}
	void rot(int x){
		int y,z,f,b;
		y=fa[x];
		z=fa[y];
		f=ch[y][0]==x;
		b=ch[x][f];
		fa[x]=z;
		fa[y]=x;
		if(b)fa[b]=y;
		ch[x][f]=y;
		ch[y][f^1]=b;
		if(ch[z][0]==y)ch[z][0]=x;
		if(ch[z][1]==y)ch[z][1]=x;
		pushup(y);
		pushup(x);
	}
	void set(int x,int v){
		d[x]=c[x]=v;
	}
	void pushdown(int x){
		if(~d[x]){
			if(ls)set(ls,d[x]);
			if(rs)set(rs,d[x]);
			d[x]=-1;
		}
	}
	bool isrt(int x){return ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x;}
	void gao(int x){
		if(!isrt(x))gao(fa[x]);
		pushdown(x);
	}
	void splay(int x){
		gao(x);
		int y,z;
		while(!isrt(x)){
			y=fa[x];
			z=fa[y];
			if(!isrt(y))rot((ch[z][0]==y)^(ch[y][0]==x)?x:y);
			rot(x);
		}
	}
	void work(){
		int i;
		for(i=1;i<=n;i++){
			fa[i]=t::fa[i][0];
			r[i]=i;
			d[i]=-1;
			siz[i]=1;
			vs[i][0]=s[i][0]=t::siz[i]-1;
		}
	}
	#define v(x) (s[x][c[x]]+siz[x])
	void access(int x,int z){
		int y,t;
		splay(x);
		y=0;
		t=0;
		while(x){
			splay(x);
			if(r[x]==z)break;
			vs[x][c[rs]]+=v(rs);
			vs[x][c[y]]-=t;
			t=v(x);
			rs=y;
			pushup(x);
			y=x;
			x=fa[x];
		}
	}
	int query(int x){
		int y,v;
		splay(x);
		y=0;
		v=c[x];
		while(x){
			splay(x);
			if(c[x]!=v)break;
			vs[x][c[rs]]+=v(rs);
			vs[x][c[y]]-=v(y);
			rs=y;
			pushup(x);
			y=x;
			x=fa[x];
		}
		return s[y][v]+siz[y];
	}
}
void modify(int x,int y,int v){
	using namespace l;
	int z=t::fa[y][0];
	if(z){
		query(z);
		splay(z);
		if(fa[z]){
			splay(fa[z]);
			vs[fa[z]][c[z]]-=v(z);
		}
		splay(y);
		vs[z][c[y]]-=v(y);
	}
	access(x,z);
	splay(x);
	set(x,v);
	if(z){
		splay(y);
		vs[z][c[y]]+=v(y);
		pushup(z);
		if(fa[z]){
			vs[fa[z]][c[z]]+=v(z);
			pushup(fa[z]);
		}
	}
}
int main(){
	int m,i,x,y,z,k;
	scanf("%d%d",&n,&m);
	t::work();
	l::work();
	while(m--){
		scanf("%d",&i);
		if(i==1){
			scanf("%d%d%d",&x,&y,&z);
			k=t::lca(x,y);
			modify(x,k,z);
			modify(y,k,z);
		}else{
			scanf("%d",&x);
			printf("%d\n",l::query(x));
		}
	}
}

[xsy2913]enos的更多相关文章

  1. 【xsy2913】 enos 动态dp

    题目大意:给你一棵 $n$个点 以 $1$为根 的树,每个点有$ 0,1,2 $三种颜色之一,初始时整棵树的颜色均为 $0$. $m$ 次操作, 每次操作形如: 1 x y c : 将 $x$到$y$ ...

  2. ng-table 简单实例

    今天用的AngularJs需要做个分页,于是用ng-table去实现,不过这个官网感觉有点坑,说的不够清楚. 下面实现了一个Demo实力,代码如下: <!DOCTYPE html> < ...

  3. AngularJs 动态加载模块和依赖

    最近项目比较忙额,白天要上班,晚上回来还需要做Angular知识点的ppt给同事,毕竟年底要辞职了,项目的后续开发还是需要有人接手的,所以就占用了晚上学习的时间.本来一直不打算写这些第三方插件的学习笔 ...

  4. ui-grid

    html代码: <html ng-app="myApp">       <head>         <meta charset="utf- ...

  5. [转]ng-grid

    本文转自:http://angular-ui.github.io/ui-grid/ Getting Started Steps for getting started (example on righ ...

  6. IGS_学习笔记10_IREP监控SOA Integration和日志设定(案例)

    20150506 Created By BaoXinjian

  7. CentOS7.1 使用资源搜集

    1.配置java环境 -openjdk* 测试 java -version 2.安装Tomcat8.0.35 点击题目可以参考源网页,但有些代码无法执行,更改如下(亲测可行): 一定要先安装java环 ...

  8. [转帖][超级少儿不宜]一氧化氮(NO),为什么亚洲人是最硬

    阴茎科学:一氧化氮(NO),为什么亚洲人是最硬 尼堪巴图鲁   ​关注他 2,911 人赞同了该文章   https://zhuanlan.zhihu.com/p/55941740 超级少儿不宜.. ...

  9. 物联网全景动态图谱2.0|PaaS物联网平台汇总(上篇)

    物联网智库 原创 物联网智库 整理发布 转载请注明来源和出处 ------   [导读]   ------ 毫无疑问,2018年物联网对行业的深度变革才刚刚开启. 物联网产业链企业的质与量将进入全面爆 ...

随机推荐

  1. win7.wifi热点

    使用本地连接上网,将网卡设为wifi热点 cmd 管理员身份运行 netsh wlan set hostednetwork mode=allow ssid=4Gtest key=12345678 网络 ...

  2. [转]Ubuntu下ROS开发环境搭建(QT+ros_qtc_plugin)

    ROS与C++入门教程-搭建开发环境(QT+ros_qtc_plugin) PS : 在“安装ros_qtc_plugin插件”这一步中,原文提到“ Ubuntu 14.04使用apt-get方式安装 ...

  3. pywin32记录备忘

    项目地址:http://sourceforge.net/projects/pywin32/ 文档地址:http://docs.activestate.com/activepython/2.7/pywi ...

  4. torch.nn.CrossEntropyLoss

    class torch.nn.CrossEntropyLoss(weight=None, size_average=True, ignore_index=-100, reduce=True) 我这里没 ...

  5. 你需要知道的12个Git高级命令【转】

    转自:http://www.linuxidc.com/Linux/2016-01/128024.htm 众所周知,Git目前已经是分布式版本控制领域的翘楚,围绕着Git形成了完整的生态圈.学习Git, ...

  6. linux编译警告 will be initialized after

    http://blog.chinaunix.net/uid-17019762-id-3152012.html 作为一个有强迫症的人,实在是受不了 warning 的存在 这个warning是由于初始化 ...

  7. 洛谷P1725 琪露诺

    传送门啦 本人第一个单调队列优化 $ dp $,不鼓励鼓励? 琪露诺这个题,$ dp $ 还是挺好想的对不,但是暴力 $ dp $ 的话会 $ TLE $ ,所以我们考虑用单调队列优化. 原题中说她只 ...

  8. 使用html+css+js实现倒计时,开启你痛苦的倒计时吧

    使用html+css+js实现倒计时,开启你痛苦的倒计时吧 效果图: 这是我痛苦的倒计时,呜呜呜 好啦,再痛苦还是要分享代码,代码如下,复制即可使用: <!DOCTYPE html> &l ...

  9. POJ 2195 Going Home(KM算法模板)

    题目链接:http://poj.org/problem?id=2195 题目大意: 给定一个N*M的地图,地图上有若干个man和house,且man与house的数量一致. man每移动一格需花费$1 ...

  10. 生成Insert语句的存储过程

    ) drop procedure [dbo].[spGenInsertSQL] GO SET QUOTED_IDENTIFIER OFF GO SET ANSI_NULLS ON GO )) as b ...