题解 「CTSC2018暴力写挂」

题目传送门

题目大意

给出两个大小为 \(n\) 的树，求出：

\[\max\{\text{depth}(x)+\text{depth}(y)-\text{depth}(\text{LCA}(x,y)-\text{depth}^{'}(\text{LCA}^{'}(x,y)))\}
\]

\(n\le 3666666\)，答案保证在 \(\text{long long}\) 范围内。

思路

边分治秒啊，终于学会了 边分树合并 了，在这里记录一下，以免后面忘掉了。

首先我们可（bu）以（ke）想（neng）到（de）一种 \(\Theta(n\log^2 n)\) 的做法，就是说我们可以先把式子化成：

\[\frac{1}{2}(\text{dist}(x,y)+\text{depth}(x)+\text{depth}(y)-2\text{depth}^{'}(\text{LCA}^{'}(x,y)))
\]

然后你发现如果不看最后那个东西的话，前面那个可以用边分治解决。然后我们发现我们可以对于第二棵树建出当前分治的点集的虚树，然后枚举某一个点作为 \(\text{LCA}^{'}(x,y)\) 。然后你就发现时间复杂度是 \(\Theta(n\log^2 n)\) ，而且常数巨大，无法通过此题。

然后我们需要引入一个叫「边分树」的东西，它跟点分树完全不一样。对于一个点我们维护一个它每一次被分治到的贡献，这道题目当中即为 \(\text{dis}(x)+\text{depth}(x)\)（\(\text{dis}(x)\) 即为 \(x\) 到分治点的距离），但是光是这样还不够，我们还需要能够更新答案，也就是说我们需要知道每次分治所属的块，这就是为什么需要是二叉树链的原因，如果某个点是它父亲的左节点，就说明他父亲是左边的块（左右其实是自己定义的，但这并不是很重要）。

考虑合并两个二叉树，对于同一相同路径走到的 \(x,y\) ，显然它们可以合成一个合法答案，直接连起来即可。

考虑如何弄到第二棵树上去，你发现如果钦定某一个点为 \(\text{LCA}\)，那么就可以用已有的点与子子树进行合并更新答案即可，不过我们还需要一个点对它自己的贡献。

时间复杂度是 \(\Theta(n\log n)\)，但是常数比较大。

\(\texttt{Code}\)

#include <bits/stdc++.h>
using namespace std;

#define Int register int
#define INF 0x7f7f7f7f7f
#define ll long long
#define MAXN 400250

char buf[1 << 25],*p1 = buf,*p2 = buf;
#define getchar() ((p1 == p2 && (p2 = (p1 = buf) + fread(buf,1,1 << 25,stdin))) ? EOF : *p1 ++)
template <typename T> inline void read (T &t){t = 0;char c = getchar();int f = 1;while (c < '0' || c > '9'){if (c == '-') f = -f;c = getchar();}while (c >= '0' && c <= '9'){t = (t << 3) + (t << 1) + c - '0';c = getchar();} t *= f;}
template <typename T,typename ... Args> inline void read (T &t,Args&... args){read (t);read (args...);}
template <typename T> inline void write (T x){if (x < 0){x = -x;putchar ('-');}if (x > 9) write (x / 10);putchar (x % 10 + '0');}

#define ls(x) tree[x].ls
#define rs(x) tree[x].rs
#define vl(x) tree[x].vl
#define vr(x) tree[x].vr

struct node{
	ll vl,vr;
	int ls,rs;
	node(){vl = vr = -INF;}
}tree[MAXN * 25];

int n,cnt,root[MAXN],las[MAXN];

namespace T1{
#define N 800250
	ll dis[N],wei[N << 1];
	int lim,ed,tot,toop = 1,to[N << 1],nxt[N << 1],vis[N << 1],siz[N],head[N];
	void Add_Edge (int u,int v,ll w){
		to[++ toop] = v,wei[toop] = w,nxt[toop] = head[u],head[u] = toop;
		to[++ toop] = u,wei[toop] = w,nxt[toop] = head[v],head[v] = toop;
	}
	void getdis (int u,int fa){for (Int i = head[u];i;i = nxt[i]) if (to[i] ^ fa) dis[to[i]] = dis[u] + wei[i],getdis (to[i],u);}
	void findedge (int u,int fa,int Siz){
		siz[u] = 1;
		for (Int i = head[u];i;i = nxt[i]){
			int v = to[i];
			if (v == fa || vis[i]) continue;
			findedge (v,u,Siz),siz[u] += siz[v];
			int tmp = max (siz[v],Siz - siz[v]);
			if (tmp < lim) lim = tmp,ed = i;
		}
	}
	void dfs (int u,int fa,ll Dis,bool kase){
		if (u <= n){
			++ tot;
			if (ls(las[u]) == -1) ls(las[u]) = tot;
			else rs(las[u]) = tot;
			if (kase == 0) ls(tot) = -1,vl(tot) = Dis + dis[u];
			else rs(tot) = -1,vr(tot) = Dis + dis[u];
			las[u] = tot;
		}
		for (Int i = head[u];i;i = nxt[i]){
			int v = to[i];if (v == fa || vis[i]) continue;
			dfs (v,u,Dis + wei[i],kase);
		}
	}
	void divide (int u,int Siz){
		if (Siz <= 1) return ;
		lim = Siz,findedge (u,0,Siz),vis[ed] = vis[ed ^ 1] = 1;
		int tx = to[ed],ty = to[ed ^ 1];
		dfs (tx,0,0,0),dfs (ty,0,wei[ed],1);
		divide (ty,Siz - siz[tx]),divide (tx,siz[tx]);
	}
	void Work (){
		for (Int i = 1;i <= n;++ i) root[i] = las[i] = ++ tot,ls(root[i]) = -1;
		getdis (1,0),divide (1,cnt);
	}
#undef N
}

namespace T2{
	ll ans,now,wei[MAXN << 1];
	int toop = 1,to[MAXN << 1],nxt[MAXN << 1],head[MAXN];
	void Add_Edge (int u,int v,ll w){
		to[++ toop] = v,wei[toop] = w,nxt[toop] = head[u],head[u] = toop;
		to[++ toop] = u,wei[toop] = w,nxt[toop] = head[v],head[v] = toop;
	}
	int Merge (int x,int y){
		if (!x || !y) return x + y;
		ans = max (ans,max (vl(x) + vr(y),vr(x) + vl(y)) - now);
		ls(x) = Merge (ls(x),ls(y)),rs(x) = Merge (rs(x),rs(y));
		vl(x) = max (vl(x),vl(y)),vr(x) = max (vr(x),vr(y));
		return x;
	}
	void dfs (int u,int fa,ll dis){
		for (Int i = head[u];i;i = nxt[i]){
			int v = to[i];if (v == fa) continue;
			dfs (v,u,dis + wei[i]),now = dis * 2,root[u] = Merge (root[u],root[v]);
		}
		ans = max (ans,(T1::dis[u] - dis) * 2);
	}
	void Work (){
		dfs (1,0,0),write (ans / 2),putchar ('\n');
	}
}

#define N 800250

ll wei[N << 1];
int toop = 1,to[N << 1],nxt[N << 1],head[N];

void Add_Edge (int u,int v,ll w){
	to[++ toop] = v,wei[toop] = w,nxt[toop] = head[u],head[u] = toop;
	to[++ toop] = u,wei[toop] = w,nxt[toop] = head[v],head[v] = toop;
}

void dfs (int u,int fa){
	for (Int i = head[u];i;i = nxt[i]){
		int v = to[i];ll w = wei[i];
		if (v == fa) continue;
		dfs (v,u);
		if (!las[u]) T1::Add_Edge (u,v,w),las[u] = u;
		else T1::Add_Edge (las[u],++ cnt,0),T1::Add_Edge (las[u] = cnt,v,w);
	}
}

signed main(){
	read (n),cnt = n;
	for (Int i = 2,u,v,w;i <= n;++ i) read (u,v,w),Add_Edge (u,v,w);
	for (Int i = 2,u,v,w;i <= n;++ i) read (u,v,w),T2::Add_Edge (u,v,w);
	dfs (1,0),T1::Work ();T2::Work ();
	return 0;
}