1.题目描述:点击打开链接
2.解题思路:本题属于树形dp问题,根据题意,我们会发现机器人在走的时候,需要在出发点,经过的点数以及是否返回到出发点这三条确定时候,才能完整的计算出走过的长度。因此,不妨令(u,k,flag)来表示从u出发,遍历了k个点(包括u自身),是否返回这样一种状态。这样,在遍历完u的子树的时候,可以用下面的方程来更新u:
dp(u,i,1)=min{dp(u,j,1)+dp(v,i-j,1)+2*dist(u,v)}(v=son(u));
dp(u,i,0)=min(dp(u,j,1)+dp(v,i-j,0)+dist(u,v),dp(u,j,0)+dp(v,i-j,1)+2*dist(u,v))(v=son(u));
第一个方程表示从u出发遍历i个点,可以看做从v出发遍历i-j个点,然后从不是v的子树出发遍历j个点,因为最后要返回u,因此dist(u,v)要计算2次。第二个方程表示从u出发遍历i个点,但是不返回u,那么有2种情况,一种是从v出发不返回,另一种是从u出发不返回。由于从非v的子树到达v的子树,必然会经过u,因此不会出发从2种子树出发均不返回的情况。
得到了dp数组后,那么只需要从大到小遍历一下dp(0,i,0)和dp(0,i,1),那么第一个小于等于x的i就是最终的答案。
3.代码: