-
递归终止条件:
- 当前节点为空,返回
null
。 - 当前节点是
p
或q
中的一个,直接返回当前节点,因为该节点可能是LCA。
- 当前节点为空,返回
-
递归左右子树:
- 分别递归左子树和右子树,获取左右子树的处理结果
left
和right
。
- 分别递归左子树和右子树,获取左右子树的处理结果
-
处理递归结果:
- 左右子树均非空:说明当前节点是
p
和q
的LCA,返回当前节点。 - 左子树结果为空:说明LCA在右子树中,返回右子树结果
right
。 - 右子树结果为空:说明LCA在左子树中,返回左子树结果
left
。
- 左右子树均非空:说明当前节点是
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (!root || root == p || root == q) return root;
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
if (left && right) return root;
if (left) return left;
if (right) return right;
return nullptr;
}
};
- 时间复杂度 O(n):每个节点最多被访问一次
- 空间复杂度 O(h):递归栈深度与树高度成正比