Description
Solution
边分治+边分树合并
这个题很多做法都是启发式合并的复杂度的,都有点卡
以前有个套路叫做线段树合并优化启发式合并,消掉一个 (log)
这个题思路类似,建出边分树,通过一些操作把它变成线段树,就可以线段树合并了
首先边分树的相关定理：
如果一棵包含 (N) 个结点的树中每个点的度均不大于 (D)，那么存在一条边，使得分出的两棵子树的结点个数在 ([N/(D+1),N*D/(D+1)])
那么边分树的深度和度数是相关的,我们只需要通过加虚点把这棵树变成二叉树就好了
回到这个题:
枚举第二棵树的 (lca) , 然后剩下的就是在第二棵树中选出在第一棵树中 (dep[x]+dep[y]-dep[lca(x,y)]) 的最大值
我们发现这个要求的东西就是 (x,y) 到根的路径的交,所以可以不考虑第一棵树的 (lca) 了
其中 (x,y) 都来自这个 (lca) 的不同子树内
每次插入一个点就像线段树那样更新就行了,不同的是你不能准确定位到这个叶子节点,你需要在 (build) 的时候顺便存一下对应关系
枚举第二棵树的 (lca) 再合并子树的边分树,顺便更新一下答案就行了
#include<bits/stdc++.h>
#define vc vector<edge>::iterator
#define pb push_back
using namespace std;
typedef long long ll;
const int N=750000,M=8510000,T=N*2;
int n,V,sz[N],sum,son[N]={N},val[T],ls[T],rs[T],dep[N],fa[T],tt=0;
ll f[N][20],dis[N],fl[M],fr[M],ans=0,D;int lm[M],rm[M],rt[N/2],id[M];
struct edge{
int x,v;
edge(){}
edge(int _x,int _v){x=_x;v=_v;}
}q[N];
vector<edge>G1[N],G2[N/2];
inline void add(int x,int y,int z,vector<edge>*G){
G[x].pb(edge(y,z));G[y].pb(edge(x,z));
}
inline void build(int x,int last){
for(vc it=G1[x].begin();it!=G1[x].end();++it){
if(it->x==last)continue;
dis[it->x]=dis[x]+it->v;build(it->x,x);
}
int l=1,r=0;edge u,v;
for(vc it=G1[x].begin();it!=G1[x].end();++it)if(it->x!=last)q[++r]=*it;
while(l+2<=r){
int o=++V;u=q[l++],v=q[l++];
add(o,u.x,u.v,G1);add(o,v.x,v.v,G1);
q[++r]=edge(o,0);
}
vector<edge>().swap(G1[x]);
while(l<=r)add(x,q[l].x,q[l].v,G1),l++;
}
inline void getdis(int x,int last,int d){
for(vc it=G1[x].begin();it!=G1[x].end();++it){
if(it->x==last || it->x==-1)continue;
f[it->x][d]=f[x][d]+it->v;getdis(it->x,x,d);
}
}
inline void getedge(int x,int last,int &ex,int &ey){
sz[x]=1;
for(int i=G1[x].size()-1,u;i>=0;i--){
if((u=G1[x][i].x)==last || u==-1)continue;
getedge(u,x,ex,ey);sz[x]+=sz[u];
if(son[u]<son[ey])ex=x,ey=u;
}
son[x]=abs(sum-2*sz[x]);
}
inline int solve(int x,int S,int d){
if(S==1){dep[x]=d;return x;}
int ex=0,ey=0,o=++V;
getdis(x,x,d);
sum=S;getedge(x,x,ex,ey);
for(vc it=G1[ex].begin();it!=G1[ex].end();++it)
if(it->x==ey){val[o]=it->v;it->x=-1;break;}
for(vc it=G1[ey].begin();it!=G1[ey].end();++it)
if(it->x==ex){it->x=-1;break;}
fa[ls[o]=solve(ex,S-sz[ey],d+1)]=o;
fa[rs[o]=solve(ey,sz[ey],d+1)]=o;
return o;
}
inline int ins(int x){
for(int i=dep[x],u=x,la=x;i>=1;i--){
id[++tt]=fa[x];fl[tt]=fr[tt]=-1ll<<60;
if(ls[fa[x]]==x)fl[tt]=max(fl[tt],dis[u]+f[u][i]),lm[tt]=la;
if(rs[fa[x]]==x)fr[tt]=max(fr[tt],dis[u]+f[u][i]),rm[tt]=la;
la=tt;x=fa[x];
}
return tt;
}
inline int merge(int x,int y){
if(!x||!y)return x+y;
ans=max(ans,(fl[x]+fr[y]+val[id[x]])/2-D);
ans=max(ans,(fl[y]+fr[x]+val[id[x]])/2-D);
fl[x]=max(fl[x],fl[y]);fr[x]=max(fr[x],fr[y]);
lm[x]=merge(lm[x],lm[y]);rm[x]=merge(rm[x],rm[y]);
return x;
}
inline void dfs(int x,int last,ll d){
rt[x]=ins(x);ans=max(ans,dis[x]-d);
for(vc it=G2[x].begin();it!=G2[x].end();++it){
if(it->x==last)continue;
dfs(it->x,x,d+it->v);D=d;
rt[x]=merge(rt[x],rt[it->x]);
}
}
int main(){
freopen("pp.in","r",stdin);
freopen("pp.out","w",stdout);
cin>>n;V=n;
int x,y,z;
for(int i=1;i<n;i++)scanf("%d%d%d",&x,&y,&z),add(x,y,z,G1);
for(int i=1;i<n;i++)scanf("%d%d%d",&x,&y,&z),add(x,y,z,G2);
build(1,1);solve(1,V,0);dfs(1,1,0);
cout<<ans<<endl;
return 0;
}