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

问题描述

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

分析

使用递归解决比较简单，但是不太好实现。
主要是要理解递归左子树与右子树，且如果两个目标结点在一条直线上，则直接返回第一个结点即可，不用再往下找。

class Solution {
public:
    TreeNode *p, *q;
    TreeNode *res;
    TreeNode* solve(TreeNode* root) {
        if (root == nullptr) {
            return nullptr;
        }
        if (root == p || root == q) {
            return root;
        }
        TreeNode* left = solve(root->left);
        TreeNode* right = solve(root->right);
        if (left && right) {return root;}
        // 返回左子树或者右子树中的目标结点，若无，返回空
        if (left) {
            return left;
        } else if (right) {
            return right;
        } else {
            return nullptr;
        }
    }

    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        this->p = p;
        this->q = q;
        TreeNode* res = solve(root);
        return res;
    }
};