最快遍历完图中重要节点 树的直径 Codeforces 328 (Div. 2) D. Super M

http://codeforces.com/problemset/problem/592/D

树的直径补:http://www.cnblogs.com/celia01/archive/2012/07/30/2615842.html

题意
给你一棵树,树上有一些重要的点,超人必须要去。
然后让你选择一个点作为起点,使得从这个点开始遍历其他的重要的点要走的距离最小
如果有多个解,就输出最小的那个


题解:
相当于建一颗包含所有重要的点,但是大小最小的虚树,然后再那个虚树里面找到字典序最小的直径
然后答案就是2*边长-直径长度就好了

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

const int MAXX=130000;
vector<int> adj[MAXX];
int dist[MAXX],mark[MAXX];
int sizee[MAXX];//该结点的子树下重要结点个数
int n,m,a,b;
void dfs(int p,int v)
{
    sizee[v]=0;
    if(mark[v]) sizee[v]=1;
    for(int j=0; j<adj[v].size(); j++)
    {
        int u=adj[v][j];
        if(u!=p)
        {
            dist[u]=dist[v]+1;//计算树的直径
            dfs(v,u);
            sizee[v]+=sizee[u];
        }
    }
}
int main ()
{
    //freopen("in.txt","r",stdin);
    cin>>n>>m;
    for(int i=0; i<n-1; i++)
    {
        cin>>a>>b;
        a--;
        b--;
        adj[a].push_back(b);
        adj[b].push_back(a);
    }
    for(int i=0; i<m; i++)
    {
        cin>>a;
        a--;
        mark[a]=1;
    }
    dfs(-1,0);
    int v=-1;
    for(int i=0; i<n; i++)
    {
        if(mark[i]&&(v==-1||dist[i]>dist[v]))
        {
            v=i;
        }
    }
    memset(dist,0,sizeof(dist));
    dfs(-1,v);//注意
    int sum=0;
    int mx=0;
    for(int i=0; i<n; i++)
    {
        if(sizee[i]>0&&m-sizee[i]>0) sum+=2;
        if(mark[i]) mx=max(dist[i],mx);
    }
    for(int i=0; i<n; i++)
    {
        if(dist[i]==mx&&i<v&&mark[i])
        {
            v=i;
            break;
        }
    }
    cout<<v+1<<'\n'<<sum-mx<<endl;
    return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值