算法 网易笔试(2020-9-12)快递问题
@author:Jingdai
@date:2020.09.15
记录一个网易笔试题,方便以后忘记的时候查看。
题目描述
原始题目是快递从0点出发,给定步数求最多经过几个快递点。
其实就是抽象后是一个树的问题,对于一个树,从根节点出发,给定步数,求最多经过几个节点(包括根节点)。
输入第一行两个数,节点数V和移动步数N,后面V-1行两个数,代表两个数的节点直连。
输入样例1:
5 2 1 0 2 1 3 2 4 3
输出样例1:
3
如图:
输入样例2:
9 5 0 1 0 2 2 6 4 2 8 1 1 3 3 7 3 5
输出样例2:
5
如图:
解题思路
如果第一次做这个题,可能比较难想,但是看了他人的题解,发现最多经过节点数分两种情况,如下公式:
就是当给定步数小于树深度时,每一步贡献一个经过节点;当给定步数大于等于树深度时,每两个贡献一个经过节点。同时考虑到算出的最大经过节点数可能会大于给的节点数,这时最大经过节点数应该就是节点数。所以公式变成下面这样:
然后发现其实整个题就非常简单了,只要求出树的深度就完了。后面的代码利用树的邻接表的层次遍历求出树的深度,完成题目。
代码
import java.util.*; public class MaxStepsSolution { public static void main(String[] args){ Scanner s = new Scanner(System.in); String tempLine = s.nextLine(); int