递归or存储父节点_236_二叉树的最近公共祖先

最新推荐文章于 2021-02-01 21:34:23 发布

小孟Tec

最新推荐文章于 2021-02-01 21:34:23 发布

阅读量164

点赞数

分类专栏： LeetCode题解文章标签：二叉树 leetcode 数据结构

本文链接：https://blog.csdn.net/m0_38024592/article/details/106035406

版权

LeetCode题解专栏收录该内容

53 篇文章 0 订阅

订阅专栏

文章目录

题目描述
思路
- 方法一：递归
- 方法二：存储父节点

题目描述

在这里插入图片描述

思路

方法一：递归

在这里插入图片描述

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

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
           
        def dfs(root, p, q):
            if not root: return False
            lson = dfs(root.left, p, q)
            rson = dfs(root.right, p, q)
            if (lson and rson) or ((root == p or root == q) and (lson or rson)):
                self.res = root
            return lson or rson or root == p or root == q

        self.res = None
        dfs(root, p, q)
        return self.res

方法二：存储父节点

用hash字典存储每个节点的父节点
然后用 hash 集合存储 p 节点所有的父节点（祖先节点） – 这里集合中需要包括 p节点自身
找 q 的父节点（祖先节点），第一次出现在 p的祖先节点集合中的节点即为结果。-- 这里判断是也需要包括 q节点自身

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

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        from collections import deque
        queue = deque()
        data_dict = {}
        queue.append(root)
        cnt = 0
        while queue:
            top = queue.popleft()
            if top.left:
                data_dict[top.left] = top
                queue.append(top.left)
            if top.right:
                data_dict[top.right] = top
                queue.append(top.right)

            if top == p:
                cnt += 1
            if top == q:
                cnt += 1
            if cnt == 2:  # 遍历完两个target node之后，提前退出循环，more efficiency
                break
        # for k, v in data_dict.items():
        #     print(k.val, v.val)
        # print(len(data_dict))

        fater_p = set()
        fater_p.add(p)
        while p in data_dict:
            fater_p.add(data_dict[p])
            p = data_dict[p]

        while True:
            if q in fater_p:
                return q
            else:
                q = data_dict[q]