题目描述

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

解题思路

p和q可能会出现两种情况：
（1）p和q分别在左、右子树中，此时返回最近公共祖先结点；
（2）p和q在同一侧子树中，此时谁为祖先结点返回谁。

回溯法

自底向上的返回信息，采用后序遍历的方式。当从左孩子或右孩子中找到p或q，则返回这个结点。这个结点再想上传递给上面的结点。

/**
 * 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) {
        // 遍历到空节点返回NULL，找到q或者p返回root。注意当出现情况二时，先找到谁，一定会先返回谁。
        if(!root || root == p || root == q)           return root;
        // 左右
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        // 中：当左和右都不为NULL，说明当前为公共祖先，返回root
        if(left && right)       return root;
        // 当一侧为零，另一侧不为零，返回孩子
        else if(!left)          return right;
        return left;
    }
};

参考文章：
236. 二叉树的最近公共祖先