二叉树中节点的第 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 是二叉树中的节点数。