[CF592D]Super M

592D:Super M

题意简述

给出一棵 n 个节点的树,其中有m个节点为标记节点。
树上相邻节点距离为 1
你被要求从任意点出发,遍历这些标记节点,求最小的总路程。

数据范围

1mn123456

思路

树形DP。
最优的方案肯定是沿着DFS序走。
每DFS到一个标记节点,就把沿路往上的路径算两遍加入贡献直到另一个标记节点。
因为我们不必回到起点,所有再做一次DFS,求出标记节点之间的最远距离,减到答案里。
这一步我们可以用类似树的直径的方法DFS两次。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
struct Node{
    int s,t,next;
}e[250010];
int head[130010],cnt;
void addedge(int s,int t)
{
    e[cnt].s=s;e[cnt].t=t;e[cnt].next=head[s];head[s]=cnt++;
    e[cnt].s=t;e[cnt].t=s;e[cnt].next=head[t];head[t]=cnt++;
}
bool pd[130010],lable[130010];
int dis[130010];
int n,m,u,v,pos,ans,tmp;
void dfs(int node,int lastfa)
{
    pd[node]=lable[node];
    for (int i=head[node];i!=-1;i=e[i].next)
        if (e[i].t!=lastfa)
        {
            dfs(e[i].t,node);
            if (pd[e[i].t])
                ans+=2;
            pd[node]|=pd[e[i].t];
        }
}
void dfs2(int node,int lastfa,int sum)
{
    dis[node]=sum;
    for (int i=head[node];i!=-1;i=e[i].next)
        if (e[i].t!=lastfa)
            dfs2(e[i].t,node,sum+1);
}
int main()
{
    scanf("%d%d",&n,&m);
    memset(head,0xff,sizeof(head));
    cnt=0;
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        addedge(u,v);
    }
    for (int i=1;i<=m;i++)
    {
        scanf("%d",&u);
        lable[u]=1;
    }
    dfs(u,u);
    dis[0]=-1;
    dfs2(u,u,0);
    for (int i=1;i<=n;i++)
        if (lable[i]&&dis[i]>dis[pos])
            pos=i;
    tmp=pos;
    dfs2(pos,pos,0);
    pos=0;
    for (int i=1;i<=n;i++)
        if (lable[i]&&dis[i]>dis[pos])
            pos=i;
    ans=ans-dis[pos];
    pos=min(pos,tmp);
    printf("%d\n%d\n",pos,ans);
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值