hdu 4607 Park Visit

最新推荐文章于 2019-10-06 08:16:52 发布

yang_bro

最新推荐文章于 2019-10-06 08:16:52 发布

阅读量834

点赞数 1

CC 4.0 BY-SA版权

分类专栏：图论

本文链接：https://blog.youkuaiyun.com/diary_yang/article/details/9464787

图论专栏收录该内容

51 篇文章

订阅专栏

本文介绍了一种计算树的直径的算法，并提供了一个完整的C++代码实现。该算法首先找到树中最远的节点，然后从该节点出发再次进行深度优先搜索以确定树的直径。对于特定的查询操作，文章还讨论了如何根据树的直径来解决相关问题。

求出树的直径d。当k<=d的时候，显然直接在直径上走就行了，当k>d的时候，便是直径d的长度加上多出来的点数*2.

#include<iostream>
#include<algorithm>
#include<vector>
#include<cstdio>
#include<cstring>
#define FF(i, a ,b) for(int i=a; i<b; i++)
#define FD(i, a ,b) for(int i=a; i>b; i--)
#define REP(i, n) for(int i=0; i<n; i++)
#define LL long long
#define CLR(a, b) memset(a, b, sizeof(a))
#define PB push_back
using namespace std;

const int maxn = 111111;
int n, m, dist, k, end;
vector<int> G[maxn];

void dfs(int x, int fa, int d)
{
    int nc = G[x].size();
    if(nc == 1)
    {
        if(d > dist)
        {
            dist = d;
            end = x;
        }
    }
    REP(i, nc)
    {
        int v = G[x][i];
        if(v != fa)
        {
            dfs(v, x, d+1);
        }
    }
}

int main()
{
    int T, a, b;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d", &n, &m);
        REP(i, n+1) G[i].clear();
        FF(i, 1, n)
        {
            scanf("%d%d", &a, &b);
            G[a].PB(b);
            G[b].PB(a);
        }
        dist = 0;
        dfs(1, -1, 0);
        dfs(end, -1, 0);
        dist++;
        while(m--)
        {
            scanf("%d", &k);
            if(k <= dist) printf("%d\n", k-1);
            else printf("%d\n", dist - 1 + (k-dist)*2);
        }
    }
    return 0;
}