树的合并（ybtoj-树上dp）

题目描述

前言

全网唯一AC！！！
妙啊
而且还是完全自己想出来的做法
开心
（APIO还是没白听）

但是思路出来后代码实现十分坎坷
建两个图分别dfs3次那个地方就算是写不明白了
代码能力需要加强

解析

定义数组dis：

dis[i]表示i在自己那棵树上所能连出的最长简单路径

不难看出：
对于将两棵树上的边x，y相连时
新的直径要么是两棵树原来一棵的直径，要么就是经过了x和y
也就是：

max3（直径A，直径B，dis[x]+dis[y]+1)

通过对两棵树分别进行dfs，我们是可以预处理出两棵树的直径和dis数组的
那么：

X=max（直径A，直径B）

就成了定值

那么我们现在考虑如何找出所有的x，y，使得二者相连的的贡献大于X
也就是：

dis[x]+dis[y]>X

我们发现：
可以把A树的dis降序排序一下
然后再枚举B的dis值
这样在A的dis序列中与B当前的dis符合条件的点分布就是单调的
可以通过二分求解
假设对于B的dis1，A的dis序列的1-p的点i满足：

dis1+disA[i]>X

那么通过dis1得到的贡献之和就是

∑(disA[i]+dis1+1)(1<=i<=p)+X*(n-p)

把西格玛拆开：

∑disA[i] (1<=i<=p)+(dis1+1)p+X(n-p)

前面的西格玛可以用前缀和预处理处来
这样我们就可以在二分求出p后O1求出dis1的贡献了
总时间复杂度：
（n+m)logn
（从复杂度可以看出来，还可以在m<n时将两棵树交换以优化复杂度，不过本题并不需要了）

代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+100;
#define ll long long
int n,m,root,mod;

struct node{
	int to,nxt;
}p[N*4];
int fi[N],cnt=-1;
void addline(int x,int y){
	p[++cnt]=(node){y,fi[x]};
	fi[x]=cnt;
//	printf("x=%d y=%d id=%d\n",x,y,cnt);
}
int fi2[N];
void addline2(int x,int y){
	p[++cnt]=(node){y,fi2[x]};
	fi2[x]=cnt;
}

int dis2A[N],disA[N],dis2B[N],disB[N];
int r=-1;
int xa,ya,xb,yb;
ll ans;
void dfsA(int x,int f){
//	printf("f=%d x=%d dis=%d\n",f,x,disA[x]);
	for(int i=fi[x];~i;i=p[i].nxt){
		int to=p[i].to;
//		printf("id=%d to=%d\n",i,to);
		if(to==f) continue;
		disA[to]=disA[x]+1;
		if(r==-1||disA[r]<disA[to]) r=to;
		dfsA(to,x);
	}
}
void dfs2A(int x,int f){
//	printf("f=%d x=%d dis=%d\n",f,x,disA[x]);
	for(int i=fi[x];~i;i=p[i].nxt){
		int to=p[i].to;
//		printf("id=%d to=%d\n",i,to);
		if(to==f) continue;
		dis2A[to]=dis2A[x]+1;
		dfs2A(to,x);
	}
}
void dfsB(int x,int f){
	for(int i=fi2[x];~i;i=p[i].nxt){
		int to=p[i].to;
		if(to==f) continue;
		disB[to]=disB[x]+1;
		if(r==-1||disB[r]<disB[to]) r=to;
		dfsB(to,x);
	}
}
void dfs2B(int x,int f){
	for(int i=fi2[x];~i;i=p[i].nxt){
		int to=p[i].to;
		if(to==f) continue;
		dis2B[to]=dis2B[x]+1;
		dfs2B(to,x);
	}
}

int X;
void solve(){
	dfsA(1,0);
	xa=r;r=-1;
	memset(disA,0,sizeof(disA));
	
	dfsA(xa,0);
	ya=r;
//	for(int i=1;i<=n;i++) printf("i=%d dis=%d\n",i,disA[i]);
	
	dfs2A(ya,0);
	for(int i=1;i<=n;i++) disA[i]=max(disA[i],dis2A[i]);
	X=disA[xa];
//	for(int i=1;i<=n;i++) printf("i=%d dis=%d\n",i,disA[i]);
	
	dfsB(1,0);
	
	xb=r;r=-1;
	memset(disB,0,sizeof(disB));
	dfsB(xb,0);
	yb=r;
	
	dfs2B(yb,0);
	for(int i=1;i<=m;i++) disB[i]=max(disB[i],dis2B[i]);
//	for(int i=1;i<=m;i++) printf("i=%d dis=%d\n",i,disB[i]);
	X=max(X,disB[xb]);
	
//	printf("%d %d %d %d X=%d\n",xa,ya,xb,yb,X);
}

bool cmp(int x,int y){return x>y;}
ll sum[N];
void compute(){
	sort(disB+1,disB+m+1,cmp);
	for(int i=1;i<=m;i++){
		sum[i]=sum[i-1]+disB[i];
	}
	for(int i=1;i<=n;i++){
		int now=disA[i];
		int st=0,ed=m;
		while(st<ed){
			int mid=(st+ed+1)>>1;
			if(disB[mid]+now+1>=X) st=mid;
			else ed=mid-1;
		}
		ans+=(ll)sum[st]+(ll)st*now+st+(ll)(m-st)*X;
	}
}
int main(){
	memset(fi,-1,sizeof(fi));
	memset(fi2,-1,sizeof(fi2));
	int a,b,c,d;
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++){
		scanf("%d%d",&a,&b);
		addline(a,b);
		addline(b,a);
	}
	for(int i=1;i<m;i++){
		scanf("%d%d",&c,&d);
		addline2(c,d);
		addline2(d,c);
	}
//	printf("dkesf=%d\n",p[0].to);
	solve();
	
	compute();
	
	printf("%lld",ans);
}
/*
3 5
1 2
1 3
1 2
1 3
3 4
4 5
*/

thanks for reading！