题目大意:带边权的树,给点一个根,问从根出发遍历某些点,所需的最小花费。
这既然是一棵树,那么从起点k到任意一个的路径都是唯一确定的(这就是树形的好处),我们可以深搜它的孩子,在过程中如果没有要访问的节点就直接返回。
否则将这条路径都标记。而且题目中可知不一定要返回到其实位置,那么可以在某个点停下。
sum[0][u]:回到u点的最短路径
sum[1][u]:不回到u点的最短路径
sum[0][u]+=len*2+sum[0][v]//1
sum[1][u]=min( len+sum[0][u]+sum[1][v], sum[1][u]+sum[0][v]+len*2 )//2
第一句很好理解,回到u就是回到v再加上u,v间路径的两倍(来回)
min中1的意思是v之前u的孩子回到u再访问v且不回到u的值,2的意思是先从v出发返回u再访问之前其他的孩子且不回来的值,取最优。
为什么?
因为依题意可知最终只会停到一个点,同时我们要消除访问先后顺序对结果的影响,所以对于v来说可能停到v的子树或u其他孩子的子树中,这个取最优即可。
树形DP就是父节点对孩子节点情况去最优。同时还因为只会有一个孩子不回到父节点,所以是可以标识出来的(每次只和一个比较,而并非和值)
#include<cstdio>
#include<cstring>
const int LMT=50002;
int sum[2][LMT],next[LMT],all,is[LMT];
struct line
{
int u,v,next,len;
}e[LMT<<1];
void init(void)
{
memset(is,0,sizeof(is));
memset(sum,0,sizeof(sum));
memset(next,-1,sizeof(next));
all=0;
}
void insert(int u,int v,int len)
{
e[all].u=u;
e[all].v=v;
e[all].len=len;
e[all].next=next[u];
next[u]=all++;
}
void dfs(int u,int pre)
{
int v,x,len,tem;
for(x=next[u];x!=-1;x=e[x].next)
if(e[x].v!=pre)
{
v=e[x].v;len=e[x].len;
dfs(v,u);//优先深搜,树形DP要依据孩子的结果对父节点取值
if(is[v])
{
tem=len+sum[0][u]+sum[1][v];
if(sum[1][u]+sum[0][v]+(len<<1)>tem)
sum[1][u]=tem;
else sum[1][u]=sum[1][u]+sum[0][v]+(len<<1);
sum[0][u]+=(len<<1)+sum[0][v];
is[u]=1;//该子树有要访问的点就标记该路径
}
}
}
int main(void)
{
int k,n,i,u,v,len,m;
while(~scanf("%d%d",&n,&k))
{
init();
for(i=1;i<n;++i)
{
scanf("%d%d%d",&u,&v,&len);
insert(u,v,len);
insert(v,u,len);//前向星
}
scanf("%d",&m);
for(i=1;i<=m;i++)
{
scanf("%d",&u);
is[u]=1;//要访问的点
}
dfs(k,-1);
printf("%d\n",sum[1][k]);
}
return 0;
}