236. 二叉樹的最近公共祖先

class Solution:def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':def dfs(root: 'TreeNode', p: 'TreeNode', q: 'TreeNode'):# 如果當(dāng)前節(jié)點(diǎn)為空，則說明 p、q 不在 node 的子樹中，不可能為公共祖先，直接返回 Noneif not root:return None# 如果當(dāng)前節(jié)點(diǎn) root 等于 p 或者 q，那么 root 就是 p、q 的最近公共祖先，直接返回 root，不需要遍歷子樹了if root == p or root == q:return root# 遞歸遍歷左子樹、右子樹，并判斷左右子樹結(jié)果node_left = dfs(root.left, p, q)node_right = dfs(root.right, p, q)# 如果左右子樹都不為空，則說明 p、q 在當(dāng)前根節(jié)點(diǎn)的兩側(cè)，當(dāng)前根節(jié)點(diǎn)就是他們的最近公共祖先if node_left and node_right:return rootreturn node_left if node_left else node_rightans = dfs(root, p, q)return ans

在遞歸函數(shù) dfs 中，我們是先判斷 root 是否等于 p 或者 q，再遍歷其左右子樹，這樣可以應(yīng)對兩種情況：1、p 或者 q 在最近公共祖先的兩側(cè)，此時(shí)由于祖先作為 root 不等于p、q，所以左右子樹都會被遍歷，從而逐級向上返回，得到祖先的左右子樹都不為空，返回 root；2、
p 或者 q 自身就是最近公共祖先，此時(shí)滿足 root 等于 p 或 q，直接返回 root 即可。

1676. 二叉樹的最近公共祖先 IV

class Solution:def lowestCommonAncestor(self, root: 'TreeNode', nodes: 'List[TreeNode]') -> 'TreeNode':def dfs(root: 'TreeNode', nodes: 'List[TreeNode]'):# 如果當(dāng)前節(jié)點(diǎn)為空，則說明 p、q 不在 node 的子樹中，不可能為公共祖先，直接返回 None，不需要遍歷子樹了if not root:return None# 如果當(dāng)前節(jié)點(diǎn) root 等于 p 或者 q，那么 root 就是 p、q 的最近公共祖先，直接返回 rootfor node in nodes:if root == node:return root# 遞歸遍歷左子樹、右子樹，并判斷左右子樹結(jié)果node_left = self.dfs(root.left, nodes)node_right = self.dfs(root.right, nodes)# 如果左右子樹都不為空，則說明 p、q 在當(dāng)前根節(jié)點(diǎn)的兩側(cè)，當(dāng)前根節(jié)點(diǎn)就是他們的最近公共祖先if node_left and node_right:return rootreturn node_left if node_left else node_rightans = dfs(root, nodes)return ans

與上一題同理，此題 p、q 換成了更多的節(jié)點(diǎn)，用列表 nodes 表示。

1644. 二叉樹的最近公共祖先 II

class Solution:def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':def dfs(root: 'TreeNode', p: 'TreeNode', q: 'TreeNode'):nonlocal pFlagnonlocal qFlag# 如果當(dāng)前節(jié)點(diǎn)為空，則說明 p、q 不在 node 的子樹中，不可能為公共祖先，直接返回 Noneif not root:return None# 遞歸遍歷左子樹、右子樹，并判斷左右子樹結(jié)果node_left = dfs(root.left, p, q)node_right = dfs(root.right, p, q)# 如果左右子樹都不為空，則說明 p、q 在當(dāng)前根節(jié)點(diǎn)的兩側(cè)，當(dāng)前根節(jié)點(diǎn)就是他們的最近公共祖先if node_left and node_right:return root# 如果當(dāng)前節(jié)點(diǎn) root 等于 p 或者 q，那么 root 就是 p、q 的最近公共祖先，返回 root，子樹也已經(jīng)被遍歷過了if root == p or root == q:if root == p:pFlag = Trueif root == q:qFlag = Truereturn rootreturn node_left if node_left else node_rightpFlag = FalseqFlag = Falseans = dfs(root, p, q)if not pFlag or not qFlag:return Nonereturn ans

這題需要用兩個(gè)標(biāo)志位記錄 p 和 q 有沒有出現(xiàn)過，如果其中之一沒出現(xiàn)過就返回 None。這題的判斷 root 是否等于 p 或 q 要放在遍歷左右子樹的后面，這是因?yàn)槿绻?root 等于 p 或 q 了然后返回，此時(shí)只有一個(gè)標(biāo)志位是變成 True 的，但我們要兩個(gè)標(biāo)志位都為 True 時(shí)才返回 ans。所以我們要先遍歷左右子樹，再判斷 root 的值，這樣就會遍歷到所有的節(jié)點(diǎn)，這種寫法在前面兩題也可以使用，但是在前面兩題的情況下是屬于效率低的。

總結(jié)

以上是生活随笔為你收集整理的二叉树最近公共祖先相关题目（Leetcode题解-Python语言）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。