Given a tree of N nodes and an integer K, each node is numbered between 1 and N. The task is to find the number of pair of ideal nodes in a tree.
A pair of nodes (a, b) is called ideal if
- a is an ancestor of b.
- And, abs(a – b) ≤ K
Input: K = 2 Output: 4 (1, 2), (1, 3), (3, 4), (3, 5) are such pairs. Input: Consider the graph in example 1 and k = 3 Output: 6 (1, 2), (1, 3), (1, 4), (3, 4), (3, 5), (3, 6) are such pairs.
Approach: First, we need to traverse the tree using DFS so we need to find the root node, the node without a parent. As we traverse each node we will store it in a data structure to keep track of all the ancestors for the next node. Before doing that, get the number of the node’s ancestors in the range [presentNode – k, presentNode + k] then add it to the total pairs.
We need a data structure which can:
- Insert a node as we traverse the tree.
- Remove a node as we return.
- Give the number of nodes within the range [presentNode – k, PresentNode + k] which were stored.
Binary indexed tree fulfills the above three operations.
We can do the above 3 operations by initializing all the index values of the BIT to 0 and then:
- Insert a node by updating the index of that node to 1.
- Remove a node by updating the index of that node to 0.
- Get the number of similar pairs of the ancestor of that node by querying for the sum of the range [presentNode – k, PresentNode + k]
Below is the implementation of the above approach:
- Find the shortest distance between any pair of two different good nodes
- Find maximum among all right nodes in Binary Tree
- Number of special nodes in an n-ary tree
- Queries to find distance between two nodes of a Binary tree
- Count the number of nodes at a given level in a tree using DFS
- Level with maximum number of nodes using DFS in a N-ary tree
- Find the path from root to the given nodes of a tree for multiple queries
- Number of leaf nodes in the subtree of every node of an n-ary tree
- Queries to find distance between two nodes of a Binary tree - O(logn) method
- Find maximum count of duplicate nodes in a Binary Search Tree
- Queries to find the maximum Xor value between X and the nodes of a given level of a perfect binary tree
- Construct XOR tree by Given leaf nodes of Perfect Binary Tree
- Find the number of distinct pairs of vertices which have a distance of exactly k in a tree
- Find number of edges that can be broken in a tree such that Bitwise OR of resulting two trees are equal
- Subtree of all nodes in a tree using DFS
If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to email@example.com. See your article appearing on the GeeksforGeeks main page and help other Geeks.
Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.
Improved By : SURENDRA_GANGWAR