树的直径最简单的做法:随便找一个的点dfs找离他最远的点p,然后再一遍dfs找距离p最远的点q。p-q即直径。
下面给出简单的小证明:
假设第一遍dfs找到了x,直径是u-v。
那么离x最远的点要么是y,z,b,a这几个点中的一个。(分别对应几种情况)
显然y和z,a不可能,因为dis(x,v)一定大于dix(y,x),dis(y,z),dis(x,a),否则,我们从u出发,离u的最远点就不是v了。
对于b,也不可能,同上,dis(x,u)一定大于dis(x,b)否则,u-v就不是直径了。
所以证明成功。
对于这一题:
我们可以先求出两个直径的端点:p,q
然后每个点查询max(dis(p,i),dis(q,i))即可。
#include <iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int M = 1e4+7;
int head[M],cnt;
struct EDGE{int to,