[UOJ347]通道

锟题x1

以下用$d_k(x,y)$表示$x,y$在第树$k$上的距离,$h_k(x)$表示$x$在树$k$上的深度

先做两棵树,即最大化$d_1(x,y)+d_2(x,y)=h_1(x)+h_1(y)-2h_1(lca)+d_2(x,y)$,其中$lca$是$x,y$在树$1$上的lca

考虑在树$1$上枚举$lca$,即是要最大化$h_1(x)+h_2(y)+d_2(x,y)$,于是我们可以对每个树$2$的点$i$建多一条边$(i,i',h_1(i))$,在dfs树$1$的同时维护(树$1$的子树中的每个$x$对应到树$2$的$x'$这些点在树$2$中的直径)即可

现在有了树$3$,考虑对其边分治,设分治到$(u,v)$,我们要查询在树$3$上跨过$(u,v)$的答案,这其实就是在两棵树的基础上加了$d_3(x,u)+d_3(y,v)+w_{u,v}$还有路径$(x,y)$必须经过$(u,v)$的约束,那么将当前分治范围内的点拿出来在树$1$上建虚树,并给$(u,v)$两边的点染不同的颜色,于是接下来我们维护不同颜色的直径,之后就和两棵树完全一样了

最后的一个小问题就是边分治的复杂度,这里直接多叉转二叉即可(对每个点新建一条$0$链)

说起来很简单,但写起来还是挺长的

#include<stdio.h>
#include<algorithm>
#include<vector>
#include<assert.h>
using namespace std;
typedef long long ll;
const int inf=2147483647;
int n;
struct tree1{
	int h[100010],nex[200010],to[200010],M;
	ll v[200010];
	void add(int a,int b,ll c){
		M++;
		to[M]=b;
		v[M]=c;
		nex[M]=h[a];
		h[a]=M;
	}
	int fa[100010],dfn[100010],mn[200010][18],dep[200010],lg[200010];
	ll d[100010];
	void dfs(int x){
		dfn[x]=++M;
		mn[M][0]=x;
		dep[x]=dep[fa[x]]+1;
		for(int i=h[x];i;i=nex[i]){
			if(to[i]!=fa[x]){
				fa[to[i]]=x;
				d[to[i]]=d[x]+v[i];
				dfs(to[i]);
				mn[++M][0]=x;
			}
		}
	}
	int qmin(int x,int y){return dep[x]<dep[y]?x:y;}
	int query(int l,int r){
		int k=lg[r-l+1];
		return qmin(mn[l][k],mn[r-(1<<k)+1][k]);
	}
	int lca(int x,int y){
		if(dfn[x]>dfn[y])swap(x,y);
		return query(dfn[x],dfn[y]);
	}
	void gao(){
		int i,j,x,y;
		ll z;
		for(i=1;i<n;i++){
			scanf("%d%d%lld",&x,&y,&z);
			add(x,y,z);
			add(y,x,z);
		}
		M=0;
		dfs(1);
		for(j=1;j<18;j++){
			for(i=1;i+(1<<j)-1<=M;i++)mn[i][j]=qmin(mn[i][j-1],mn[i+(1<<(j-1))][j-1]);
		}
		for(i=2;i<=M;i++)lg[i]=lg[i>>1]+1;
	}
}t1;
struct tree2{
	int h[100010],nex[200010],to[200010],M;
	ll v[200010];
	void add(int a,int b,ll c){
		M++;
		to[M]=b;
		v[M]=c;
		nex[M]=h[a];
		h[a]=M;
	}
	int fa[100010],lg[200010],in[100010];
	ll d[100010],mn[200010][19];
	void dfs(int x){
		in[x]=++M;
		mn[M][0]=d[x];
		for(int i=h[x];i;i=nex[i]){
			if(to[i]!=fa[x]){
				fa[to[i]]=x;
				d[to[i]]=d[x]+v[i];
				dfs(to[i]);
				mn[++M][0]=d[x];
			}
		}
	}
	ll query(int l,int r){
		int k=lg[r-l+1];
		return min(mn[l][k],mn[r-(1<<k)+1][k]);
	}
	ll dis(int x,int y){
		if(in[x]>in[y])swap(x,y);
		return d[x]+d[y]-2*query(in[x],in[y]);
	}
	void gao(){
		int i,j,x,y;
		ll z;
		for(i=1;i<n;i++){
			scanf("%d%d%lld",&x,&y,&z);
			add(x,y,z);
			add(y,x,z);
		}
		M=0;
		dfs(1);
		for(j=1;j<19;j++){
			for(i=1;i+(1<<j)-1<=M;i++)mn[i][j]=min(mn[i][j-1],mn[i+(1<<(j-1))][j-1]);
		}
		for(i=2;i<=M;i++)lg[i]=lg[i>>1]+1;
	}
}t2;
struct pr{
	int to;
	ll v;
	pr(int a=0,ll b=0){to=a;v=b;}
};
struct tree3{
	int h[200010],nex[400010],to[400010],M;
	ll v[400010];
	void ins(int a,int b,ll c){
		M++;
		to[M]=b;
		v[M]=c;
		nex[M]=h[a];
		h[a]=M;
	}
	void add(int a,int b,ll c){
		ins(a,b,c);
		ins(b,a,c);
	}
	vector<pr>g[100010];
	int N;
	void dfs(int fa,int x){
		int p;
		vector<pr>::iterator it;
		if(g[x].size()){
			p=0;
			for(it=g[x].begin();it!=g[x].end();it++){
				if(it->to!=fa){
					if(p){
						N++;
						add(p,N,0);
						add(N,it->to,it->v);
						p=N;
					}else{
						add(x,it->to,it->v);
						p=x;
					}
				}
			}
		}
		for(it=g[x].begin();it!=g[x].end();it++){
			if(it->to!=fa)dfs(x,it->to);
		}
	}
	void gao(){
		int i,x,y;
		ll z;
		for(i=1;i<n;i++){
			scanf("%d%d%lld",&x,&y,&z);
			g[x].push_back(pr(y,z));
			g[y].push_back(pr(x,z));
		}
		N=n;
		M=1;
		dfs(0,1);
	}
}t3;
ll dis[200010],ad;
ll get(int x,int y){
	return t2.dis(x,y)+t1.d[x]+t1.d[y]+dis[x]+dis[y]+ad;
}
struct dia{
	int x,y;
	dia(int a=0,int b=0){x=a;y=b;}
};
ll max(dia a,dia b){
	if(a.x==0||b.x==0)return 0;
	return max(max(get(a.x,b.x),get(a.x,b.y)),max(get(a.y,b.x),get(a.y,b.y)));
}
dia operator+(dia a,dia b){
	if(a.x==0)return b;
	if(b.x==0)return a;
	int x,y;
	ll mx=0,t;
	#define ch(u,v) t=get(u,v);\
					if(t>mx){\
						x=u;\
						y=v;\
						mx=t;\
					}
	ch(a.x,a.y)
	ch(b.x,b.y)
	ch(a.x,b.x)
	ch(a.x,b.y)
	ch(a.y,b.x)
	ch(a.y,b.y)
	return dia(x,y);
}
struct pd{
	dia x,y;
	pd(dia a=0,dia b=0){x=a;y=b;}
	dia&operator[](int k){return k?x:y;}
};
pd operator+(pd a,pd b){return pd(a.x+b.x,a.y+b.y);}
ll max(pd a,pd b){return max(max(a.x,b.y),max(a.y,b.x));}
ll ans;
bool vis[400010];
int siz[200010],p[200010],M;
void dfs1(int fa,int x){
	if(x<=n)p[++M]=x;
	siz[x]=1;
	for(int i=t3.h[x];i;i=t3.nex[i]){
		if(!vis[i]&&t3.to[i]!=fa){
			dfs1(x,t3.to[i]);
			siz[x]+=siz[t3.to[i]];
		}
	}
}
int al,mn,cn;
void dfs2(int fa,int x){
	for(int i=t3.h[x];i;i=t3.nex[i]){
		if(!vis[i]&&t3.to[i]!=fa){
			dfs2(x,t3.to[i]);
			if(abs(al-2*siz[t3.to[i]])<mn){
				mn=abs(al-2*siz[t3.to[i]]);
				cn=i;
			}
		}
	}
}
struct vtree{
	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 col[100010];
	pd dfs(int x){
		pd d,t;
		d[col[x]]=dia(x,x);
		for(int i=h[x];i;i=nex[i]){
			t=dfs(to[i]);
			ans=max(ans,max(d,t)-2*t1.d[x]);
			d=d+t;
		}
		h[x]=0;
		return d;
	}
}vt;
bool cmp(int x,int y){return t1.dfn[x]<t1.dfn[y];}
int st[100010],tp;
void insert(int x){
	if(!tp){
		st[++tp]=x;
		return;
	}
	int l=t1.lca(x,st[tp]);
	while(tp>1&&t1.dep[st[tp-1]]>t1.dep[l]){
		vt.add(st[tp-1],st[tp]);
		tp--;
	}
	if(t1.dep[st[tp]]>t1.dep[l]){
		vt.add(l,st[tp]);
		tp--;
	}
	if(t1.dep[st[tp]]<t1.dep[l])st[++tp]=l;
	st[++tp]=x;
}
void build(){
	int i;
	sort(p+1,p+M+1,cmp);
	tp=0;
	for(i=1;i<=M;i++)insert(p[i]);
	for(i=1;i<tp;i++)vt.add(st[i],st[i+1]);
	if(st[1]!=1)vt.add(1,st[1]);
}
void dfs3(int fa,int x,int f){
	if(x<=n)vt.col[x]=f;
	for(int i=t3.h[x];i;i=t3.nex[i]){
		if(!vis[i]&&t3.to[i]!=fa){
			dis[t3.to[i]]=dis[x]+t3.v[i];
			dfs3(x,t3.to[i],f);
		}
	}
}
void solve(int x){
	int y;
	M=0;
	dfs1(0,x);
	al=siz[x];
	mn=inf;
	cn=0;
	dfs2(0,x);
	if(cn==0)return;
	vis[cn]=vis[cn^1]=1;
	x=t3.to[cn];
	y=t3.to[cn^1];
	build();
	dis[x]=dis[y]=0;
	dfs3(0,x,0);
	dfs3(0,y,1);
	ad=t3.v[cn];
	vt.dfs(1);
	vt.M=0;
	solve(x);
	solve(y);
}
int main(){
	scanf("%d",&n);
	t1.gao();
	t2.gao();
	t3.gao();
	solve(1);
	printf("%lld",ans);
}

猜你喜欢

转载自www.cnblogs.com/jefflyy/p/9638734.html
347