📌  相关文章
📜  二叉树中节点的第 K 个祖先 |设置 2

📅  最后修改于: 2022-05-13 01:57:16.471000             🧑  作者: Mango

二叉树中节点的第 K 个祖先 |设置 2

给定一棵二叉树,其中节点从 1 到 n 编号。给定一个节点和一个正整数 K。我们必须在二叉树中打印给定节点的第 K 个祖先。如果不存在任何这样的祖先,则打印 -1。
例如,在下面给出的二叉树中,5 的第二个祖先是 1。节点 5 的第三个祖先将是 -1。

我们在上一篇文章中讨论了针对此问题的基于 BFS 的解决方案。如果您仔细观察该解决方案,您会发现基本方法是首先找到节点,然后回溯到第 k 个父节点。同样的事情可以使用递归 DFS 来完成,而无需使用额外的数组。
使用 DFS 的想法是首先在树中找到给定的节点,然后回溯 k 次以到达第 k 个祖先,一旦我们到达第 k 个父节点,我们将简单地打印该节点并返回 NULL。
下面是上述想法的实现:

C++
/* C++ program to calculate Kth ancestor of given node */
#include 
using namespace std;
 
// A Binary Tree Node
struct Node
{
    int data;
    struct Node *left, *right;
};
 
// temporary node to keep track of Node returned
// from previous recursive call during backtrack
Node* temp = NULL;
 
// recursive function to calculate Kth ancestor
Node* kthAncestorDFS(Node *root, int node , int &k)
{  
    // Base case
    if (!root)
        return NULL;
     
    if (root->data == node||
       (temp =  kthAncestorDFS(root->left,node,k)) ||
       (temp =  kthAncestorDFS(root->right,node,k)))
    {  
        if (k > 0)       
            k--;
         
        else if (k == 0)
        {
            // print the kth ancestor
            cout<<"Kth ancestor is: "<data;
             
            // return NULL to stop further backtracking
            return NULL;
        }
         
        // return current node to previous call
        return root;
    }
}
 
// Utility function to create a new tree node
Node* newNode(int data)
{
    Node *temp = new Node;
    temp->data = data;
    temp->left = temp->right = NULL;
    return temp;
}
 
// Driver program to test above functions
int main()
{
    // Let us create binary tree shown in above diagram
    Node *root = newNode(1);
    root->left = newNode(2);
    root->right = newNode(3);
    root->left->left = newNode(4);
    root->left->right = newNode(5);
 
    int k = 2;
    int node = 5;
 
    // print kth ancestor of given node
    Node* parent = kthAncestorDFS(root,node,k);
     
    // check if parent is not NULL, it means
    // there is no Kth ancestor of the node
    if (parent)
        cout << "-1";
     
    return 0;
}


Java
// Java program to calculate Kth ancestor of given node
class Solution
{
 
// A Binary Tree Node
static class Node
{
    int data;
    Node left, right;
};
 
// temporary node to keep track of Node returned
// from previous recursive call during backtrack
static Node temp = null;
static int k;
 
// recursive function to calculate Kth ancestor
static Node kthAncestorDFS(Node root, int node )
{
    // Base case
    if (root == null)
        return null;
     
    if (root.data == node||
    (temp = kthAncestorDFS(root.left,node)) != null ||
    (temp = kthAncestorDFS(root.right,node)) != null)
    {
        if (k > 0)    
            k--;
         
        else if (k == 0)
        {
            // print the kth ancestor
            System.out.print("Kth ancestor is: "+root.data);
             
            // return null to stop further backtracking
            return null;
        }
         
        // return current node to previous call
        return root;
    }
    return null;
}
 
// Utility function to create a new tree node
static Node newNode(int data)
{
    Node temp = new Node();
    temp.data = data;
    temp.left = temp.right = null;
    return temp;
}
 
// Driver code
public static void main(String args[])
{
    // Let us create binary tree shown in above diagram
    Node root = newNode(1);
    root.left = newNode(2);
    root.right = newNode(3);
    root.left.left = newNode(4);
    root.left.right = newNode(5);
 
    k = 2;
    int node = 5;
 
    // print kth ancestor of given node
    Node parent = kthAncestorDFS(root,node);
     
    // check if parent is not null, it means
    // there is no Kth ancestor of the node
    if (parent != null)
        System.out.println("-1");
}
}
 
// This code is contributed by Arnab Kundu


Python3
""" Python3 program to calculate Kth
    ancestor of given node """
 
# A Binary Tree Node
# Utility function to create a new tree node
class newNode:
 
    # Constructor to create a new node
    def __init__(self, data):
        self.data = data
        self.left = None
        self.right = None
 
# recursive function to calculate
# Kth ancestor
def kthAncestorDFS(root, node, k):
     
    # Base case
    if (not root):
        return None
     
    if (root.data == node or
       (kthAncestorDFS(root.left, node, k)) or
       (kthAncestorDFS(root.right, node, k))):
         
        if (k[0] > 0):
            k[0] -= 1
         
        else if (k[0] == 0):
             
            # print the kth ancestor
            print("Kth ancestor is:", root.data)
             
            # return None to stop further
            # backtracking
            return None
             
        # return current node to previous call
        return root
     
# Driver Code
if __name__ == '__main__':
    root = newNode(1)
    root.left = newNode(2)
    root.right = newNode(3)
    root.left.left = newNode(4)
    root.left.right = newNode(5)
 
    k = [2]
    node = 5
 
    # print kth ancestor of given node
    parent = kthAncestorDFS(root,node,k)
     
    # check if parent is not None, it means
    # there is no Kth ancestor of the node
    if (parent):
        print("-1")
 
# This code is contributed
# by SHUBHAMSINGH10


C#
// C# program to calculate Kth ancestor of given node
using System;
     
class GFG
{
 
// A Binary Tree Node
public class Node
{
    public int data;
    public Node left, right;
};
 
// temporary node to keep track of Node returned
// from previous recursive call during backtrack
static Node temp = null;
static int k;
 
// recursive function to calculate Kth ancestor
static Node kthAncestorDFS(Node root, int node )
{
    // Base case
    if (root == null)
        return null;
     
    if (root.data == node||
    (temp = kthAncestorDFS(root.left,node)) != null ||
    (temp = kthAncestorDFS(root.right,node)) != null)
    {
        if (k > 0)    
            k--;
         
        else if (k == 0)
        {
            // print the kth ancestor
            Console.Write("Kth ancestor is: "+root.data);
             
            // return null to stop further backtracking
            return null;
        }
         
        // return current node to previous call
        return root;
    }
    return null;
}
 
// Utility function to create a new tree node
static Node newNode(int data)
{
    Node temp = new Node();
    temp.data = data;
    temp.left = temp.right = null;
    return temp;
}
 
// Driver code
public static void Main(String []args)
{
    // Let us create binary tree shown in above diagram
    Node root = newNode(1);
    root.left = newNode(2);
    root.right = newNode(3);
    root.left.left = newNode(4);
    root.left.right = newNode(5);
 
    k = 2;
    int node = 5;
 
    // print kth ancestor of given node
    Node parent = kthAncestorDFS(root,node);
     
    // check if parent is not null, it means
    // there is no Kth ancestor of the node
    if (parent != null)
        Console.WriteLine("-1");
}
}
 
// This code is contributed by 29AjayKumar


Javascript


输出:

Kth ancestor is: 1

时间复杂度:O(n),其中 n 是二叉树中的节点数。