1530. 好叶子节点对的数量-优快云博客

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right

class Solution:
    def countPairs(self, root: Optional[TreeNode], distance: int) -> int:
        self.ans = 0

        def dfs(node):
            if not node:
                return [0] * distance
            if not node.left and not node.right:
                dist = [0] * distance
                dist[0] = 1
                return dist

            left = dfs(node.left)
            right = dfs(node.right)

            # 统计左右子树之间的好叶子节点对
            for i in range(distance):
                for j in range(distance):
                    if i + j + 2 <= distance:
                        self.ans += left[i] * right[j]

            # 构建向上传递的距离数组
            merged = [0] * distance
            for i in range(distance - 1):
                merged[i + 1] = left[i] + right[i]
            return merged

        dfs(root)
        return self.ans

📊 分析与比较

方法	时间复杂度	空间复杂度	是否超时	可读性
暴力法	O(n²)（枚举所有叶子节点对）	O(n)	会超时	简单但低效
DFS + 剪枝（当前方法）	O(n * d²)（每个节点比较 d² 次）	O(d)	✅ 不超时	✅ 高效且直观

说明：

n 是二叉树中的节点数。
d 是 distance 的值（最大为10，根据题目约束）。
由于 distance 很小，d² 是常量，因此 DFS 方法在实际中表现优异。

🧪 更多示例

示例 2：

root = [1]
distance = 1

输出：0
说明： 只有一个节点，无叶子对。

示例 3：

root = [1,2,3,4,5,null,6]
distance = 3

树结构如下：

叶子节点为：[4, 5, 6]。满足条件的对有：

4 & 5：路径为 4→2→5，距离2
5 & 6：路径为 5→2→1→3→6，距离4（不满足）
4 & 6：路径为 4→2→1→3→6，距离4（不满足）

输出：1

✨ 总结

本题考查的是树的递归遍历与路径问题。通过深度优先搜索，我们可以有效地自底向上收集叶子节点的距离信息，并在遍历过程中完成对符合条件的叶子节点对的统计。

🔑 技巧回顾：

自底向上的 DFS 可用于统计叶子到当前节点的路径。
利用数组 dist 优雅处理距离分布。
注意剪枝：数组只保留到 distance 的信息，防止爆炸式递归。