Codeforces Round #328 (Div. 2) D. Super M

题目#328(Div.2)D题

题目大意

给你一颗树， 和一些重要的节点，超人要遍历每一个重要的点
起点任意，求要走的路径值最小为多少，多组解输出起点最小的那个。

解题思路

建立一颗连通所有重要节点的最小树， 设路径和为e，则任意一个重要节点
遍历所有重要节点再返回的路径和为2e，那么最优解就是从一个重要节点开始
遍历所有节点并在离他最远的那个节点结束。最远距离即为此树的直径L。
所以答案为2e-L。

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <stack>
#include <bitset>
#define INF 1000000005
#define eps 1e-12
#define PI acos(-1.0)
#define LL long long

using namespace std;

const int maxn = 2e5;
const int mod = 1000000007;

vector<int> G[maxn];
bool is[maxn], vis[maxn];
int ans = 0, tot, d[maxn], D;

void dfs(int x)
{
    vis[x] = true; int tmp = tot;
    if(is[x])tot++;
    for(int i = 0; i < G[x].size(); i++){
        int e = G[x][i];
        if(!vis[e]) dfs(e);
    }
    if(tot > tmp) ans++;
}

void dfs1(int x, int deep)
{
    vis[x] = true; d[x] = deep;
    for(int i = 0; i < G[x].size(); i++){
        int e = G[x][i];
        if(!vis[e]) dfs1(e, deep+1);
    }
}

void dfs2(int x, int deep)
{
    vis[x] = true; d[x] = max(d[x], deep);
    for(int i = 0; i < G[x].size(); i++){
        int e = G[x][i];
        if(!vis[e]) dfs2(e, deep+1);
    }
}

void solve()
{
    int n, m;
    scanf("%d%d", &n, &m);
    memset(is, false, sizeof(is));
    for(int i = 1; i < n; i++){
        int x, y;
        scanf("%d%d", &x, &y);
        G[x].push_back(y);
        G[y].push_back(x);
    }
    int x = 0, y;
    for(int i = 0; i < m; i++){
        int xx; scanf("%d", &xx);
        is[xx] = true;
        if(!i) x = xx; // x可以为任意一个重要节点
    }
    memset(vis, false, sizeof(vis));
    dfs(x); ans = ans * 2 - 2; //找到总长度

    memset(vis, false, sizeof(vis));
    dfs1(x, 0);
    y = x;
    for(int i = 1; i <= n; i++){
        if(is[i] && d[i] > d[y]) y = i;//找到最长路径的一个端点
    }
    memset(vis, false, sizeof(vis)); dfs1(y,0);
    for(int i = 1; i <= n; i++) if(is[i] &&d[i] > d[x]) x = i;//找到最长路径的另一个端点
    D = d[x]; ans -= D;

    memset(d, false, sizeof(d));
    memset(vis, false, sizeof(vis)); dfs2(x, 0); //找到每个节点的最大深度d[i];
    memset(vis, false, sizeof(vis)); dfs2(y, 0);

    for(int i = 1; i <= n; i++){
        if(is[i] && d[i] == D){
            printf("%d\n%d\n", i, ans);
            return;
        }
    }
}

int main()
{
    //freopen("in.txt", "r", stdin);
    solve();
    return 0;
}