68-II. El ancestro común más cercano del árbol de búsqueda binaria (simple)

Descripción del Título:

Dado un árbol binario, encuentre el ancestro común más cercano de dos nodos especificados en el árbol. La definición del ancestro común más cercano en la Enciclopedia Baidu es: "Para dos nodos pyq del árbol enraizado T, el ancestro común más cercano se expresa como un nodo x, de modo que x es el ancestro de pyq y la profundidad de x es lo más grande posible (Un nodo también puede ser su propio antepasado) ".

Por ejemplo, dado el siguiente árbol binario: root = [3,5,1,6,2,0,8, null, null, 7,4]

Ejemplo 1: Entrada: raíz = [3,5,1,6,2,0,8, nulo, nulo, 7,4], p = 5, q = 1
Salida: 3
Explicación: El más cercano del nodo 5 y del nodo 1 El ancestro común es el nodo 3.

Ejemplo 2:

Entrada: raíz = [3,5,1,6,2,0,8, nulo, nulo, 7,4], p = 5, q = 4
Salida: 5
Explicación: El ancestro común más cercano del nodo 5 y el nodo 4 es Nodo 5. Porque, por definición, el nodo ancestro común más cercano puede ser el propio nodo.

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        # 递归
        # 如果p、q的值和根节点相同,那么直接返回根节点
        # 如果root是None,直接返回None
        # 在root这条线上继续寻找,如果左边没找到,说明在右边
        # 如果右边没找到说明在左边
        if not root:return None
        if p.val ==root.val or q.val==root.val:return root
        l=self.lowestCommonAncestor(root.left,p,q)
        r=self.lowestCommonAncestor(root.right,p,q)

        return root if l and r else l or r

 

Supongo que te gusta

Origin blog.csdn.net/weixin_38664232/article/details/105013624
Recomendado
Clasificación