【dtoj#4605】城镇

传送门

题目描述

L国一共有 $N$ 座城镇，开始时它们两两不连通。
L国计划依次建造 $N-1$ 条道路，把所有城镇连通起来。每建完一条道路，你需要回答这条道路所在连通块内距离最远的两座城镇之间的距离。
两座城镇之间的距离定义为从一座走到另一座所需要经过的最少道路数。

输入格式

第一行一个整数 $N$，表示城镇的数量。
接下来 $N-1$ 行，每行两个整数 $a_i,b_i$ 表示接下来建的道路连通的两座城镇。
保证 $N-1$ 条道路能够使所有城镇连通。

输出格式

输出 $N-1$ 行，每行一个整数表示建完第 $i$ 条道路后的答案。

样例输入

5
3 5
3 4
1 2
1 3

样例输出

1
2
1
3

数据范围与提示

对于 $20\%$ 的数据，$n\le 300$
对于 $50\%$ 的数据， $n\le 2000$
对于另外 $20\%$ 的数据，保证 $b_i=i+1,a_i\le i$;
对于 $100\%$ 的数据，$n\le 300000$

题解

读完题目脑子里第一个反应就该是直径（树上最远两点距离），然而……我太弱力，刚做这题时还不知道直径……
想到直径，50分的解法就出来了：每添加一条边暴力求直径。

求直径的方法：
1 两遍bfs(dfs)，先从任意节点出发遍历找到最远点，再从这个最远点出发找到它的最远点，那么这两点距离就是直径
2 树形dp。
· 对于节点x，求出以节点x为根能够到达的最远节点距离d[x]，则有d[x]=max{d[y]+dis[x][y]}。
· 对于节点x，求出经过节点x的最长链长度f[x]，则树的直径就是max{f[x]}，设j<i，则有f[x]=max{d[yi]+d[yj]+dis[x][yi]+dis[x][yj]}。（y都是x的子节点）
· 当更新d[x]的遍历将要枚举到i，d[x]=max{d[yj]+dis[x][yj]}。因此不需要两层循环遍历i和j，只需先更新f[x]=max{d[x]+d[yi]+dis[x][yi]}，再更新d[x]=max{d[yi]+dis[x][yi]}即可。

然而对于两棵已知直径的树，当它们连接，新的直径只可能有6种情况：（AB为一棵树的直径两端，CD为另一棵树的）

A-B,A-C,A-D,B-C,B-D,C-D
于是首先预处理出整棵树上的点的距离，配合并查集，维护每个点所在连通块的直径两端，每加入一条边对6种情况贪心并更新直径两端即可。
代码跟网上题解对拍过了，不能说是我写的，因为改错改到抓狂orz但为了方便整理还是放上来

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
int aa[300005],bb[300005],n,dep[300005],jump[300005][20],fa[300005],d1[300005],d2[300005];
vector <int> rds[300005];
void dfs(int nw,int come){
	jump[nw][0]=come;
	for(int i=0;i<rds[nw].size();i++){
		if(rds[nw][i]==come) continue;
		dep[rds[nw][i]]=dep[nw]+1;
		dfs(rds[nw][i],nw);
	}
}
int lca(int a,int b){
	if(dep[a]!=dep[b]){
		if(dep[a]<dep[b]) swap(a,b);//a更深 
		int cj=dep[a]-dep[b];
		for(int qq=0;qq<=19;qq++) if(cj&(1<<qq)) a=jump[a][qq];
	}
	if(a==b) return a;
	for(int qq=19;qq>=0;qq--) if(jump[a][qq]!=jump[b][qq]) a=jump[a][qq],b=jump[b][qq];
	if(a==b) return a; 
	return jump[a][0];
}
int dis(int a,int b){
	return dep[a]+dep[b]-dep[lca(a,b)]*2;
}
int findfa(int x){
	if(fa[x]==x) return x;
	else return fa[x]=findfa(fa[x]);
}
void cmp(int &f,int &s,int &res,int a,int b){
	if(dis(a,b)>res){
		res=dis(a,b);
		f=a;
		s=b;
	}
}
int ana(int a,int b){
	int res=0,faa=findfa(a),fab=findfa(b);
	fa[faa]=fab;
	int w=d1[faa],x=d2[faa],y=d1[fab],z=d2[fab];
	cmp(d1[fab],d2[fab],res,w,x);
	cmp(d1[fab],d2[fab],res,w,y);
	cmp(d1[fab],d2[fab],res,w,z);
	cmp(d1[fab],d2[fab],res,x,y);
	cmp(d1[fab],d2[fab],res,x,z);
	cmp(d1[fab],d2[fab],res,y,z);
	return res;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n-1;i++){
		scanf("%d%d",&aa[i],&bb[i]);
		fa[i]=d1[i]=d2[i]=i;
		rds[aa[i]].push_back(bb[i]);
		rds[bb[i]].push_back(aa[i]);
	}
	fa[n]=d1[n]=d2[n]=n;
	dfs(1,1);
	for(int j=1;j<=19;j++)
		for(int i=1;i<=n;i++)
			jump[i][j]=jump[jump[i][j-1]][j-1];
	for(int i=1;i<=n-1;i++){
		printf("%d\n",ana(aa[i],bb[i]));
	}
	return 0;
}

Thanks!