leetcode美团真题练习03

meituan-005. 小美的区域会议

题目描述

小美是美团总部的高管，她想要召集一些美团的区域负责人来开会，已知美团的业务区域划分可以用一棵树来表示，树上有 n 个节点，每个节点分别代表美团的一个业务区域，每一个区域有一个负责人，这个负责人的级别为 A[i]
已知小美召集人员开会必须满足以下几个条件：

• 小美召集的负责人所在的区域必须构成一个非空的连通的图，即选取树上的一个连通子图。
• .这些负责人中，级别最高的和级别最低的相差不超过 k 。

请问小美有多少种召集负责人的方式，当且仅当选取的集合不同时我们就认为两种方式不同。由于方案数可能非常大，所以请对 10^9+7 取模。

输入：

• 输入第一行包含两个整数 n 和 k ，表示区域的数量，和不能超过的级别。
• 接下来有 n-1 行，每行有两个正整数 a 和 b ，表示 a 号区域和 b 号区域有一条边。
• 最后一行有 n 个整数，第 i 个整数表示 i 号区域负责人的级别。
  输出：
• 输出仅包含一个整数，表示可以选择的方案数对 10^9+7 取模之后的结果。

示例：
输入：
5 1
1 2
2 3
3 4
4 5
2 2 2 2 2
输出：15
解释：显然一个区域的方案有 {1}，{2}，{3}，{4}，{5}，两个区域的方案有 4 个，三个区域的方案有 3 个，四个区域的方案有 2 个，五个区域的方案有 1 个，共 15 个。

解题思路

复盘

这题我的思路是以每个节点$i$为根节点，dfs遍历，记录从根节点到$j$的级别差记录到$dp[i][j]$.
之后遍历dp找小于$k$的值.
后来发现我这只是从$i$到$j$的单条不重复路径, 对于重复路径的情况没考虑.

题解

参考题解枚举最小值+DFS, 以当前节点$i$为权限最小的根节点建树, 记录可能的集合数$root=\prod (nod{e}_i+$