191110-Luogu-P4556-雨天的尾巴

题目传送门

题目背景改编

己亥年十月，小F累计切掉了好多好多 $NOI/NOI+/CTSC$ 难度的题。有一种自豪和成就感在小F内心油然而生，他高兴地走下座位，说：“唉，看来我是连树的遍历都搞不来的最菜的菜鸡，你们都是巨神！” “ FAKE! ” 大家都知道小F又开始装弱了。他装弱的时候会在机房（可以看作一棵树）从一个点走到一个点。由于重力过大，每次装弱都在机房的地砖上留下了“五彩缤纷”（形容种类为 $1$~$10^5$ ）的痕迹。他为了减少影响力，于是问你所有个点最多的痕迹是什么，他顺便就把树的遍历“搞懂”了。

题目描述

一棵有 $n$ 个节点的树， $m$ 次操作，每次选择两点 $x,y$ ，在它们的最短路经过的所有点做一个种类为 $z$ 的标记，求最后每个点被标记次数最多的是那种标记。

输入格式

第一行两个正整数 $n,m(1\le n,m\le 10^5)$ 。
接下来 $n-1$ 行，每行两个数 $a,b(1\le a,b\le n)$ ，表示 $a,b$ 点间有一条无向边。
再接下来 $m$ 行，每行三个数 $x,y(1\le x,y\le n),z(1\le z\le 10^5)$ ，描述不同的路径标记。

输出格式

$n$ 行，第 $i$ 行 $1$ 个整数，表示第 $i$ 个节点最多的记号，如果有多种标记数量相同，输出编号最小的。如果没有标记，则该行输出 $0$ 。

样例输入

5 3
1 2
3 1
3 4
5 3
2 3 3
1 5 2
3 3 3

样例输出

2
3
3
0
2

解析

背景：对于线段树我还要加强练习，顺便我也可以复习大量的模板，用了一个晚上过了这道题。

$PART1$

编写预处理及 最近公共祖先 的模板，可以使用倍增法，注意不要把数组开得太大，空间（ $128M$ ）卡的很死。

void Init(int u,int fa){
	dep[u]=dep[fa]+1;
	for(int i=0;i<=17;i++)f[u][i+1]=f[f[u][i]][i];
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		f[v][0]=u;Init(v,u);
	}
}
int Lca(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	for(int i=18;i>=0;i--){
		if(dep[f[x][i]]>=dep[y])x=f[x][i];
		if(x==y)return x;
	}
	for(int i=18;i>=0;i--)if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
	return f[x][0];
}

$PART2$

记录路径从每一个点到另一个点的标记，可以装桶（形态犹如 $b[i][z]$ 第 $i$ 个点标记为 $z$ 的出现了多少次）。如果一个一个“火”速记录的话肯定会超时，因此 树上差分 粉墨登场。在开始 $x$ 和结束 $y$ 的位置标记这个记号 $+1$ ，它们的 $LCA$ 及 $f[LCA][0]$ 这个记号 $-1$ 最后遍历的时候累加。这样经过预算要占用 $37G$ 的空间，显然需要换一个办法。暴力做法：

	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&s[i],&t[i],&a[i]);
		int uu=Lca(s[i],t[i]);
		b[s[i]][a[i]]++;
		b[t[i]][a[i]]++;
		b[uu][a[i]]--;
		b[f[uu][0]][a[i]]--;
	}

$PART3$

我们可以把记录每个点 $b$ 数组改为一棵权值线段树，记录值域上出现的次数，经过检验，$6666666$ 可以。（由于值域不算大，我们可以不离散化）
具体做法：开始就构建 $n$ 个根节点代表题中树每个节点，准备从根开始谱新篇。底层的每个节点下标为数值。每次迎来路径的时候，在种类编号的下标处进行单点修改，有动态开点的操作。到达叶子节点时加上 $num$ 出现次数，递归时节点编号的最大（出现次数最多，相等且最小）等于左右儿子最大值取 $\max$ 。

void Insert(int &p,int pos,int l,int r,int num){//主席树节点编号，要改的值下标，值域左、右端，+1/-1
	if(!p)p=++sz;//没有就开点
	if(l==r){tr[p].maxx=make_pair(tr[p].maxx.first+num,-pos);return;}//到底了，某个种类的标记出现次数+1/-1，对应下标
	int mid=(l+r)>>1;
	if(pos<=mid)Insert(tr[p].lc,pos,l,mid,num);//递归左子树
	else Insert(tr[p].rc,pos,mid+1,r,num);//递归右子树
	tr[p].maxx=max(tr[tr[p].lc].maxx,tr[tr[p].rc].maxx);//权值线段树每个节点从下层统计最大值
}
...
Init(1,0);int s,t,a,uu;
for(int i=1;i<=m;i++){
	scanf("%d%d%d",&s,&t,&a);
	uu=Lca(s,t);
	Insert(rt[s],a,1,100000,1),Insert(rt[t],a,1,100000,1);//起点终点+1
	Insert(rt[uu],a,1,100000,-1),Insert(rt[f[uu][0]],a,1,100000,-1);//LCA及faLCA-1
}

$PART4$

在最后一次遍历的时候，我们要合并那个根以它的子节点为线段树根的线段树，边合并边记录最大值出现次数（及时更新）和那个值是多少，可以用 $pair<int,int>$ 存，方便比大小。
具体做法：往不空的方向，从把左儿子和左儿子合并，右儿子和右儿子合并，每个数的出现次数相加。

int Merge(int p,int q,int l,int r){
	if(!p)return q;//这边空了去那一边
	if(!q)return p;//那边空了去这一边
	if(l==r){tr[p].maxx.first+=tr[q].maxx.first;return p;}//到头了
	int mid=(l+r)>>1;
	tr[p].lc=Merge(tr[p].lc,tr[q].lc,l,mid);//合并左儿子
	tr[p].rc=Merge(tr[p].rc,tr[q].rc,mid+1,r);//合并右儿子
	tr[p].maxx=max(tr[tr[p].lc].maxx,tr[tr[p].rc].maxx);//刷新极大值
	return p;//结束
}

$PART5$

遍历的时候递归时合并线段树，每个节点的答案是（必然是合并子节点中的最大值产生）出现最多的对子的另一半，最后输出就 圆满成功。

void Bl(int u,int fa){
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		Bl(v,u);
		rt[u]=Merge(rt[u],rt[v],1,100000);//把v并到u
	}
	ans[u]=-tr[rt[u]].maxx.second;//另一半
}

Code

#include<bits/stdc++.h>
using namespace std;
int head[100002],f[200002][19],dep[200002],tot,n,m,sz,ans[100002],rt[100002];
struct EDGE{int nxt,to;}e[200002];//邻接表
struct node{int lc,rc;pair<int,int>maxx;}tr[6666666];//权值线段树
inline void Add(int x,int y){e[++tot]=(EDGE){head[x],y};head[x]=tot;}
void Init(int u,int fa){
	dep[u]=dep[fa]+1;
	for(int i=0;i<=17;i++)f[u][i+1]=f[f[u][i]][i];
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		f[v][0]=u;Init(v,u);
	}
}
int Lca(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	for(int i=18;i>=0;i--){
		if(dep[f[x][i]]>=dep[y])x=f[x][i];
		if(x==y)return x;
	}
	for(int i=18;i>=0;i--)if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
	return f[x][0];
}
void Insert(int &p,int pos,int l,int r,int num){
	if(!p)p=++sz;
	if(l==r){tr[p].maxx=make_pair(tr[p].maxx.first+num,-pos);return;}
	int mid=(l+r)>>1;
	if(pos<=mid)Insert(tr[p].lc,pos,l,mid,num);
	else Insert(tr[p].rc,pos,mid+1,r,num);
	tr[p].maxx=max(tr[tr[p].lc].maxx,tr[tr[p].rc].maxx);
}
int Merge(int p,int q,int l,int r){
	if(!p)return q;
	if(!q)return p;
	if(l==r){tr[p].maxx.first+=tr[q].maxx.first;return p; }
	int mid=(l+r)>>1;
	tr[p].lc=Merge(tr[p].lc,tr[q].lc,l,mid);
	tr[p].rc=Merge(tr[p].rc,tr[q].rc,mid+1,r);
	tr[p].maxx=max(tr[tr[p].lc].maxx,tr[tr[p].rc].maxx);
	return p;
}
void Bl(int u,int fa){
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		Bl(v,u);
		rt[u]=Merge(rt[u],rt[v],1,100000);
	}
	ans[u]=-tr[rt[u]].maxx.second;
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n-1;i++){
		int x,y;scanf("%d%d",&x,&y);
		Add(x,y),Add(y,x);
	}
	Init(1,0);int s,t,a,uu;
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&s,&t,&a);
		uu=Lca(s,t);
		Insert(rt[s],a,1,100000,1),Insert(rt[t],a,1,100000,1);
		Insert(rt[uu],a,1,100000,-1),Insert(rt[f[uu][0]],a,1,100000,-1);
	}
	Bl(1,0);
	for(int i=1;i<=n;i++)printf("%d\n",ans[i]);
	return 0;
}//https://www.luogu.org/problem/P4556 权值线段树+动态开点与合并+最近公共祖先+树上差分 

压行还是挺好的，最后就是注意常数问题。恭喜通过了一道题，加油！

预祝大家身体健康，CSP取得好成绩。