Talentd
FleetCode

1530. Number of Good Leaf Nodes Pairs

Medium71.9% acceptance3 companies

You are given the root of a binary tree and an integer distance. A pair of two different leaf nodes of a binary tree is said to be good if the length of the shortest path between them is less than or equal to distance.

Return the number of good leaf node pairs in the tree.

Example 1:

Input: root = [1,2,3,null,4], distance = 3
Output: 1
Explanation: The leaf nodes of the tree are 3 and 4 and the length of the shortest path between them is 3. This is the only good pair.

Example 2:

Input: root = [1,2,3,4,5,6,7], distance = 3
Output: 2
Explanation: The good pairs are [4,5] and [6,7] with shortest path = 2. The pair [4,6] is not good because the length of ther shortest path between them is 4.

Example 3:

Input: root = [7,1,4,6,null,5,3,null,null,null,null,null,2], distance = 3
Output: 1
Explanation: The only good pair is [2,5].

Constraints:

  • The number of nodes in the tree is in the range [1, 210].
  • 1 <= Node.val <= 100
  • 1 <= distance <= 10

Asked at 3 companies

Hints open one at a time

Hint 1
Start DFS from each leaf node. stop the DFS when the number of steps done > distance.
Hint 2
If you reach another leaf node within distance steps, add 1 to the answer.
Hint 3
Note that all pairs will be counted twice so divide the answer by 2.
20:22
Still stuck?Read 3 approaches, or watch a walkthrough.
Open solution
Loading editor…
1# Definition for a binary tree node.
2# class TreeNode:
3# def __init__(self, val=0, left=None, right=None):
4# self.val = val
5# self.left = left
6# self.right = right
7class Solution:
8 def countPairs(self, root: Optional[TreeNode], distance: int) -> int:
9
Starter codePython 3Spaces: 4
Examples
Input
[1,2,3,null,4]
3
Input
[1,2,3,4,5,6,7]
3
Input
[7,1,4,6,null,5,3,null,null,null,null,null,2]
3
Your code is saved on this device.Solution