题意:给出n个点,以及n-1条边,求逛k个点的最小距离。
思路:求出这颗树的直径(即相距最远的两点间的距离),若k小于等于树的直径上的点的个数,则答案就是k-1,若大于了树的直径,则除了走树的直径外的边都是走两遍。
所以重点就变成了求树的直径,以下为树的直径求法及证明。
转:(由于没有找到文章源头,就不贴地址了)
树的直径(Diameter)是指树上的最长简单路。
直径的求法:两遍BFS (or DFS)
任选一点u为起点,对树进行BFS遍历,找出离u最远的点v
以v为起点,再进行BFS遍历,找出离v最远的点w。则v到w的路径长度即为树的直径
*简单证明
于是原问题可以在O(E)时间内求出
关键在于证明第一次遍历的正确性,也就是对于任意点u,距离它最远的点v一定是最长路的一端。
如果u在最长路上,那么v一定是最长路的一端。可以用反证法:假设v不是最长路的一端,则存在另一点v’使得(u→v’)是最长路的一部分,于是len(u→v’) > len(u→v)。但这与条件“v是距u最远的点”矛盾。
如果u不在最长路上,则u到其距最远点v的路与最长路一定有一交点c,且(c→v)与最长路的后半段重合(why?),即v一定是最长路的一端
因为是树是连通的,所以u必有一条路径c和最长路径L相交,len(c)>=1,L被分为两部分,一部分l1,一部分l2
假设第一次dfs过后,所求最长路径lu端不在L上,那么len(lu)>=len(c)+len(l1)(l1,l2对称,取l1或者l2都一样)
len(l2+c+lu)>len(l1+l2),矛盾.
#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <map>
#include <vector>
#include <cmath>
#include <stack>
#include <queue>
#include <cstdlib>
#include <algorithm>
using namespace std;
typedef __int64 int64;
typedef long long ll;
#define M 100005
#define max_inf 0x7f7f7f7f
#define min_inf 0x80808080
#define mod 1000000007
struct node
{
int ev , next;
}edge[2*M];
int n , m , head[M] , tot , dis , poi;
void Addedge(int s , int e)
{
edge[tot].ev = e;
edge[tot].next = head[s];
head[s] = tot++;
}
void Dfs(int s , int fa , int index)
{
int i;
for (i = head[s] ; i != -1 ; i = edge[i].next)
{
int v = edge[i].ev;
if (v == fa)continue;
Dfs(v,s,index+1);
}
if (index > dis)
{
dis = index;
poi = s;
}
}
int main()
{
int t , i;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
tot = 0;
memset(head , -1 , sizeof head);
for (i = 1 ; i < n ; i++)
{
int s , e;
scanf("%d%d",&s,&e);
Addedge(s,e);
Addedge(e,s);
}
dis = 0;
Dfs(1,0,0);
dis = 0;
Dfs(poi,0,0);
while (m--)
{
int k;
scanf("%d",&k);
if (k <= dis+1)
printf("%d\n",k-1);
else
printf("%d\n",dis+(k-(dis+1))*2);
}
}
return 0;
}