Iam_Sandeep

Kth ancestor of a node in binary tree

Mar 23rd, 2022 (edited)
159
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.44 KB | None | 0 0
  1. link="https://www.codingninjas.com/codestudio/problems/kth-ancestor-of-a-node-in-binary-tree_842561?leftPanelTab=0"
  2. def findKthAncestor(root, tar, k):
  3.     ans=[None]
  4.     def dfs(x):
  5.         if not x:return None
  6.         if x.data==tar:return 1
  7.         a=dfs(x.left)
  8.         b=dfs(x.right)
  9.         if (a==k or b==k) and not ans[0]:ans[0]=x
  10.         return 1+a if a else (1+b if b else None)
  11.     dfs(root)
  12.     return ans[0].data if ans[0] else -1
  13.  
Add Comment
Please, Sign In to add comment