标签：CF592D 个点 int siz dfs dep tag 直径 Super

[CF592D] Super M - 树的直径

Description

有一个有 n 个节点的树，其中有 m 个点被标记。现在从任意点开始，去遍历这 m 个点，问你遍历这 m个点的最少路程（边的权值为1）。

Solution

答案显然是这个标记点构成的虚树大小减去虚树直径

当然我们不需要真的建虚树，从一个标记点开始 DFS，考察子树内是否有标记点即可

#include <bits/stdc++.h>
using namespace std;

#define int long long

int n, m;

const int N = 1e6 + 5;

vector<int> g[N];
int dep[N], siz[N], tag[N];

void dfs(int p, int from)
{
    siz[p] = 0;
    for (int q : g[p])
    {
        if (q == from)
            continue;
        dep[q] = dep[p] + 1;
        dfs(q, p);
        siz[p] += siz[q];
    }
    if (tag[p] || siz[p])
        ++siz[p];
}

signed main()
{
    ios::sync_with_stdio(false);

    cin >> n >> m;
    for (int i = 1; i < n; i++)
    {
        int x, y;
        cin >> x >> y;
        g[x].push_back(y);
        g[y].push_back(x);
    }

    int s = 1e9;
    for (int i = 1; i <= m; i++)
    {
        int x;
        cin >> x;
        s = min(s, x);
        tag[x] = 1;
    }

    if (m == 1)
    {
        cout << s << endl
             << 0 << endl;
    }
    else
    {
        dep[s] = 1;
        dfs(s, 0);
        for (int i = 1; i <= n; i++)
            dep[i] *= tag[i];
        int a = 0;
        for (int i = 1; i <= n; i++)
            if (dep[i] > dep[a])
                a = i;
        dep[a] = 1;
        dfs(a, 0);
        for (int i = 1; i <= n; i++)
            dep[i] *= tag[i];
        int len = *max_element(dep + 1, dep + n + 1);
        int b = 0;
        for (int i = 1; i <= n; i++)
            if (dep[i] > dep[b])
                b = i;
        cout << min(a, b) << endl
             << 2 * (siz[a] - 1) - len + 1 << endl;
    }
}

标签：CF592D,个点,int,siz,dfs,dep,tag,直径,Super
来源： https://www.cnblogs.com/mollnn/p/14638772.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。