友情支持
如果您觉得这个笔记对您有所帮助，看在D瓜哥码这么多字的辛苦上，请友情支持一下，D瓜哥感激不尽，😜
|
|
有些打赏的朋友希望可以加个好友，欢迎关注D 瓜哥的微信公众号，这样就可以通过公众号的回复直接给我发信息。
公众号的微信号是: jikerizhi。因为众所周知的原因，有时图片加载不出来。 如果图片加载不出来可以直接通过搜索微信号来查找我的公众号。 |
1530. 好叶子节点对的数量
给你二叉树的根节点 root 和一个整数 distance 。
如果二叉树中两个 叶 节点之间的 最短路径长度 小于或者等于 distance ，那它们就可以构成一组 好叶子节点对 。
返回树中 好叶子节点对的数量 。
示例 1：
输入：root = [1,2,3,null,4], distance = 3 输出：1 解释：树的叶节点是 3 和 4 ，它们之间的最短路径的长度是 3 。这是唯一的好叶子节点对。
示例 2：
输入：root = [1,2,3,4,5,6,7], distance = 3 输出：2 解释：好叶子节点对为 [4,5] 和 [6,7] ，最短路径长度都是 2 。但是叶子节点对 [4,6] 不满足要求，因为它们之间的最短路径长度为 4 。
示例 3：
输入：root = [7,1,4,6,null,5,3,null,null,null,null,null,2], distance = 3 输出：1 解释：唯一的好叶子节点对是 [2,5] 。
示例 4：
输入：root = [100], distance = 1 输出：0
示例 5：
输入：root = [1,1,1], distance = 2 输出：1
提示：
-
tree的节点数在[1, 210]范围内。 -
每个节点的值都在
[1, 100]之间。 -
1 <= distance <= 10
思路分析
深度优先遍历
-
一刷
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
/**
* 思路OK，通过 74 / 113 个测试用例。再检查一下哪里有问题？！
*
* @author D瓜哥 · https://www.diguage.com
* @since 2025-05-03 08:07:40
*/
int result = 0;
int distance;
public int countPairs(TreeNode root, int distance) {
this.distance = distance;
dfs(root);
return result;
}
private Map<Integer, List<TreeNode>> dfs(TreeNode root) {
if (root == null) {
return new HashMap<>();
}
if (root.left == null && root.right == null) {
return new HashMap<>(Map.of(0, new ArrayList<>(Arrays.asList(root))));
}
Map<Integer, List<TreeNode>> left = dfs(root.left);
Map<Integer, List<TreeNode>> right = dfs(root.right);
Map<Integer, List<TreeNode>> map = new HashMap<>();
for (Map.Entry<Integer, List<TreeNode>> entry : left.entrySet()) {
int key = entry.getKey();
List<TreeNode> nodes = entry.getValue();
map.put(key + 1, nodes);
int diffNum = distance - (key + 1) - 1;
for (int i = diffNum; i >= 0 && !right.isEmpty(); i--) {
List<TreeNode> diff = right.remove(i);
if (diff != null) {
result += diff.size() * nodes.size();
List<TreeNode> exist = map.getOrDefault(i + 1, new ArrayList<>());
exist.addAll(diff);
map.put(i + 1, exist);
}
}
}
for (Map.Entry<Integer, List<TreeNode>> entry : right.entrySet()) {
int key = entry.getKey();
List<TreeNode> nodes = entry.getValue();
List<TreeNode> exist = map.getOrDefault(key + 1, new ArrayList<>());
exist.addAll(nodes);
map.put(key + 1, exist);
int diffNum1 = distance - (key + 1) - 1;
for (int i = diffNum1; i >= 0 && !left.isEmpty(); i--) {
List<TreeNode> diff = left.get(i);
if (diff != null) {
result += diff.size() * nodes.size();
}
}
}
return map;
}